結果

問題 No.1308 ジャンプビーコン
ユーザー semisagisemisagi
提出日時 2020-12-05 01:35:05
言語 Swift
(5.10.0)
結果
AC  
実行時間 3,036 ms / 4,000 ms
コード長 3,416 bytes
コンパイル時間 2,524 ms
コンパイル使用メモリ 132,328 KB
実行使用メモリ 151,040 KB
最終ジャッジ日時 2024-09-15 09:52:32
合計ジャッジ時間 58,005 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge6
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 9 ms
9,216 KB
testcase_01 AC 9 ms
9,344 KB
testcase_02 AC 9 ms
9,344 KB
testcase_03 AC 9 ms
9,216 KB
testcase_04 AC 9 ms
9,344 KB
testcase_05 AC 9 ms
9,344 KB
testcase_06 AC 9 ms
9,216 KB
testcase_07 AC 9 ms
9,344 KB
testcase_08 AC 10 ms
9,344 KB
testcase_09 AC 10 ms
9,344 KB
testcase_10 AC 10 ms
9,216 KB
testcase_11 AC 11 ms
9,344 KB
testcase_12 AC 10 ms
9,344 KB
testcase_13 AC 84 ms
14,592 KB
testcase_14 AC 84 ms
14,592 KB
testcase_15 AC 84 ms
14,360 KB
testcase_16 AC 84 ms
14,576 KB
testcase_17 AC 84 ms
14,368 KB
testcase_18 AC 2,847 ms
150,608 KB
testcase_19 AC 2,818 ms
150,728 KB
testcase_20 AC 2,857 ms
150,476 KB
testcase_21 AC 2,851 ms
150,784 KB
testcase_22 AC 2,876 ms
150,528 KB
testcase_23 AC 9 ms
9,344 KB
testcase_24 AC 11 ms
9,472 KB
testcase_25 AC 11 ms
9,344 KB
testcase_26 AC 2,674 ms
150,664 KB
testcase_27 AC 2,785 ms
150,476 KB
testcase_28 AC 2,658 ms
150,752 KB
testcase_29 AC 2,871 ms
150,912 KB
testcase_30 AC 2,802 ms
151,040 KB
testcase_31 AC 2,015 ms
150,784 KB
testcase_32 AC 2,837 ms
151,040 KB
testcase_33 AC 3,036 ms
151,040 KB
testcase_34 AC 2,695 ms
150,528 KB
testcase_35 AC 2,695 ms
150,784 KB
testcase_36 AC 2,752 ms
150,528 KB
testcase_37 AC 2,710 ms
150,528 KB
testcase_38 AC 2,902 ms
150,784 KB
testcase_39 AC 2,915 ms
150,784 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

struct Scanner {
    private var elements = [String]()
    private var index = 0

    mutating func peek() -> String {
        while elements.count == index {
            elements = readLine()!.split(separator: " ").map(String.init)
            index = 0
        }
        return elements[index]
    }

    mutating func next() -> String {
        defer { index += 1 }
        return peek()
    }

    mutating func nextInt() -> Int {
        return Int(next())!
    }

    mutating func nextInts(_ n: Int) -> [Int] {
        return (0 ..< n).map { _ in nextInt() }
    }

    mutating func nextDouble() -> Double {
        return Double(next())!
    }
}

struct Array2D<Element> {
    let n1: Int
    let n2: Int
    private var elements: [Element]

    init(repeating: Element, _ n1: Int, _ n2: Int) {
        self.n1 = n1
        self.n2 = n2
        self.elements = [Element](repeating: repeating, count: n1 * n2)
    }

    subscript(i: Int, j: Int) -> Element {
        get {
            elements[i * n2 + j]
        }
        set {
            elements[i * n2 + j] = newValue
        }
    }
}

var scanner = Scanner()
let N = scanner.nextInt()
let Q = scanner.nextInt()
let C = scanner.nextInt()

var tree = [[(Int, Int)]](repeating: [], count: N)

for _ in 0 ..< N - 1 {
    let u = scanner.nextInt() - 1
    let v = scanner.nextInt() - 1
    let l = scanner.nextInt()
    tree[u].append((v, l))
    tree[v].append((u, l))
}

var distance = Array2D<Int>(repeating: 0, N, N)

func dfs(u: Int, parent: Int?, root: Int) {
    for (v, w) in tree[u] where v != parent {
        distance[root, v] = distance[root, u] + w
        dfs(u: v, parent: u, root: root)
    }
}
for i in 0 ..< N {
    dfs(u: i, parent: nil, root: i)
}

let X = scanner.nextInts(Q).map { $0 - 1 }

let inf = Int.max / 2

var dp = Array2D<Int>(repeating: inf, Q, N)
dp[0, X[0]] = 0

var prefix = [[Int]]()
var suffix = [[Int]]()

for i in 0 ..< N {
    prefix.append([Int](repeating: inf, count: tree[i].count + 1))
    suffix.append([Int](repeating: inf, count: tree[i].count + 1))
}

for i in 0 ..< Q - 1 {
    var dp2 = [Int](repeating: inf, count: N)
    func dfs(_ u: Int, _ parent: Int?) {
        dp2[u] = dp[i, u]
        for (v, w) in tree[u] where v != parent {
            dfs(v, u)
            dp2[u] = min(dp2[u], dp2[v] + w)
        }
    }
    func dfs2(_ u: Int, _ parent: Int?) {
        prefix[u][0] = dp[i, u]
        for i in tree[u].indices {
            let (v, w) = tree[u][i]
            prefix[u][i + 1] = min(prefix[u][i], dp2[v] + w)
        }
        suffix[u][tree[u].count] = dp[i, u]
        for i in tree[u].indices.reversed() {
            let (v, w) = tree[u][i]
            suffix[u][i] = min(suffix[u][i + 1], dp2[v] + w)
        }
        dp[i + 1, u] = min(dp[i + 1, u], suffix[u][0] + C + distance[u, X[i + 1]])
        for i in tree[u].indices {
            let v = tree[u][i].0
            guard v != parent else { continue }
            dp2[u] = min(prefix[u][i], suffix[u][i + 1])
            dfs2(v, u)
        }
    }
    dfs(0, nil)
    dfs2(0, nil)

    for j in 0 ..< N where dp[i, j] != inf {
        dp[i + 1, j] = min(dp[i + 1, j], dp[i, j] + distance[X[i], X[i + 1]])
    }

    let m = (0 ..< N).map { dp[i, $0] }.min()!
    for j in 0 ..< N {
        dp[i + 1, j] = min(dp[i + 1, j], m + distance[X[i], j] + distance[j, X[i + 1]])
    }
}

print((0 ..< N).map { dp[Q - 1, $0] }.min()!)
0