結果

問題 No.994 ばらばらコイン
ユーザー Topology1225Topology1225
提出日時 2020-03-14 18:41:38
言語 Python3
(3.12.2 + numpy 1.26.4 + scipy 1.12.0)
結果
WA  
実行時間 -
コード長 1,029 bytes
コンパイル時間 225 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 41,368 KB
最終ジャッジ日時 2024-11-23 21:22:48
合計ジャッジ時間 5,934 ms
ジャッジサーバーID
(参考情報)
judge2 / judge3
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 WA -
testcase_01 AC 29 ms
10,624 KB
testcase_02 AC 435 ms
35,340 KB
testcase_03 AC 474 ms
41,368 KB
testcase_04 AC 466 ms
41,312 KB
testcase_05 AC 243 ms
25,712 KB
testcase_06 WA -
testcase_07 AC 462 ms
41,184 KB
testcase_08 AC 307 ms
25,624 KB
testcase_09 AC 435 ms
39,832 KB
testcase_10 AC 335 ms
30,276 KB
testcase_11 AC 371 ms
32,524 KB
testcase_12 AC 382 ms
33,716 KB
testcase_13 WA -
testcase_14 WA -
testcase_15 WA -
testcase_16 AC 28 ms
10,624 KB
testcase_17 AC 28 ms
10,624 KB
testcase_18 AC 28 ms
10,752 KB
testcase_19 AC 27 ms
10,752 KB
testcase_20 AC 27 ms
10,752 KB
testcase_21 AC 28 ms
10,624 KB
testcase_22 WA -
権限があれば一括ダウンロードができます

ソースコード

diff #

class NTree(object):
    def __init__(self):
        self.count = 0

    def constructor(self,N, K, edges, num):
        self.N = int(N)
        self.K = int(K)
        self.edges = edges
        self.num = num 
        if self.N < self.K:
            print(-1)
            exit()

    def run(self):
        sorted_num = dict(sorted(self.num.items(), key=lambda x:x[0]))
        K = self.K
        for key, value in sorted_num.items():
            if value >= K:
                self.count += K
                break
            else:
                self.count += value
                K = K - value
        print(self.count-1)

if __name__=="__main__":
    N, K = input().split(" ")
    edges= list()
    num  = dict()
    for i in range(1, int(N)):
        edge = list(map(int, input().split(" ")))
        edges.append(edge)
        if not edge[0] in num.keys():
            num[edge[0]]  = 1
        else:
            num[edge[0]] += 1
    func = NTree()
    func.constructor(N=N, K=K, edges=edges, num=num)
    func.run()
0