結果

問題 No.482 あなたの名は
ユーザー バカらっく
提出日時 2018-06-11 14:33:22
言語 Swift
(6.0.3)
結果
AC  
実行時間 116 ms / 2,000 ms
コード長 1,267 bytes
コンパイル時間 1,170 ms
コンパイル使用メモリ 133,920 KB
実行使用メモリ 20,328 KB
最終ジャッジ日時 2024-11-30 12:29:37
合計ジャッジ時間 4,151 ms
ジャッジサーバーID
(参考情報)
judge5 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 28
権限があれば一括ダウンロードができます

ソースコード

diff #

class Proc {
    private let itemCount:Int
    private let moveCount:Int64
    private let items:[Int]
    private var group:[Int]
    init() {
        let inpt = readLine()!.split(separator: " ").map{ Int64($0)!}
        self.itemCount = Int(inpt[0])
        self.moveCount = inpt[1]
        items = readLine()!.split(separator: " ").map{ Int($0)! - 1 }
        self.group = (0..<self.itemCount).map{ _ in 0 }
    }

    public func doProc() {
        var groupId = 0
        var cnt:Int64 = 0
        for i in 0..<self.itemCount {
            if items[i] == i {
                continue
            }
            if group[i] != 0 {
                continue
            }
            groupId += 1
            group[i] = groupId
            let tmp = cntGroupItem(i)
            cnt += Int64(tmp - 1)
        }
        var ans = "NO"
        if moveCount >= cnt && (Int64(moveCount) - Int64(cnt)) % Int64(2) == 0 {
            ans = "YES"
        }
        print(ans)
    }

    private func cntGroupItem(_ target:Int) -> Int {
        let grp = group[target]
        let next = items[target]

        var ans = 1
        if group[next] == 0 {
            group[next] = grp
            ans += cntGroupItem(next)
        }
        return ans
    }
}

Proc().doProc()
0