提出 #17978677

---

ソースコード 拡げる

func readInt4(line: Int = #line, file: String = #file) -> (Int, Int, Int, Int) {
    guard let string = readLine() else {
        preconditionFailure("No input (at line \(line) in \(file))")
    }
    let values: [Int] = string.split(separator: " ").map { part in
        guard let value = Int(part) else {
            preconditionFailure("Illegal input value: \(string) (at line \(line) in \(file))")
        }
        return value
    }
    precondition(values.count == 4, "Illegal number of input values: count = \(values.count), values = \(values) (at line \(line) in \(file))")
    return (values[0], values[1], values[2], values[3])
}

func readInt2(line: Int = #line, file: String = #file) -> (Int, Int) {
    guard let string = readLine() else {
        preconditionFailure("No input (at line \(line) in \(file))")
    }
    let values: [Int] = string.split(separator: " ").map { part in
        guard let value = Int(part) else {
            preconditionFailure("Illegal input value: \(string) (at line \(line) in \(file))")
        }
        return value
    }
    precondition(values.count == 2, "Illegal number of input values: count = \(values.count), values = \(values) (at line \(line) in \(file))")
    return (values[0], values[1])
}

struct Array2D<Element>: Sequence {
    let width: Int
    let height: Int
    private(set) var elements: [Element]
    init(width: Int, height: Int, elements: [Element]) {
        precondition(elements.count == width * height)
        self.width = width
        self.height = height
        self.elements = elements
    }
    init(width: Int, height: Int, element: Element) {
        self.init(width: width, height: height, elements: [Element](repeating: element, count: width * height))
    }
    var count: Int { width * height }
    var xRange: Range<Int> { 0 ..< width }
    var yRange: Range<Int> { 0 ..< height }
    private func indexAt(x: Int, y: Int) -> Int {
        precondition(xRange.contains(x))
        precondition(yRange.contains(y))
        return y * width + x
    }
    subscript(x: Int, y: Int) -> Element {
        get { elements[indexAt(x: x, y: y)] }
        set { elements[indexAt(x: x, y: y)] = newValue }
    }
    func makeIterator() -> IndexingIterator<[Element]> {
        elements.makeIterator()
    }
    func map<T>(_ transform: (Element) throws -> T) rethrows -> Array2D<T> {
        try Array2D<T>(width: width, height: height, elements: elements.map(transform))
    }
}

let (h, w, n, m) = readInt4()

enum Cell {
    case light
    case block
    case none
}

var map: Array2D<Cell> = .init(width: w, height: h, element: .none)
var lightMap: Array2D<Bool> = .init(width: w, height: h, element: false)

for _ in 0 ..< n {
    let (b, a) = readInt2()
    map[a - 1, b - 1] = .light
}
for _ in 0 ..< m {
    let (d, c) = readInt2()
    map[c - 1, d - 1] = .block
}

for y in map.yRange {
    var isLighting = false
    for x in map.xRange {
        switch map[x, y] {
        case .light:
            isLighting = true
        case .block:
            isLighting = false
        case .none:
            break
        }
        if isLighting {
            lightMap[x, y] = true
        }
    }
}

for y in map.yRange {
    var isLighting = false
    for x in map.xRange.reversed() {
        switch map[x, y] {
        case .light:
            isLighting = true
        case .block:
            isLighting = false
        case .none:
            break
        }
        if isLighting {
            lightMap[x, y] = true
        }
    }
}

for x in map.xRange {
    var isLighting = false
    for y in map.yRange {
        switch map[x, y] {
        case .light:
            isLighting = true
        case .block:
            isLighting = false
        case .none:
            break
        }
        if isLighting {
            lightMap[x, y] = true
        }
    }
}

for x in map.xRange {
    var isLighting = false
    for y in map.yRange.reversed() {
        switch map[x, y] {
        case .light:
            isLighting = true
        case .block:
            isLighting = false
        case .none:
            break
        }
        if isLighting {
            lightMap[x, y] = true
        }
    }
}

let count = lightMap.reduce(into: 0) {
    if $1 { $0 += 1 }
}
print(count)

提出情報

ジャッジ結果