結果

問題 No.1477 Lamps on Graph
コンテスト
ユーザー tatt61880
提出日時 2021-06-06 09:32:31
言語 Kuin
(KuinC++ v.2021.9.17)
結果
AC  
実行時間 462 ms / 2,000 ms
コード長 1,528 bytes
コンパイル時間 2,854 ms
コンパイル使用メモリ 150,656 KB
実行使用メモリ 29,568 KB
最終ジャッジ日時 2024-09-16 12:34:00
合計ジャッジ時間 12,336 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 38
権限があれば一括ダウンロードができます

ソースコード

diff # 

func main()
	var n: int :: cui@inputInt()
	var m: int :: cui@inputInt()
	var a: []int :: #[n]int
	for i(0, n - 1)
		do a[i] :: cui@inputInt()
	end for
	
	var uvs: []list<int> :: #[n]list<int>
	for i(0, n - 1)
		do uvs[i] :: #list<int>
	end for
	
	for(1, m)
		var u: int :: cui@inputInt() - 1
		var v: int :: cui@inputInt() - 1
		do uvs[u].add(v)
		do uvs[v].add(u)
	end for
	
	var c: []bool :: #[n]bool
	var k: int :: cui@inputInt()
	for i(0, k - 1)
		do c[cui@inputInt() - 1] :: true
	end for
	
	class ValIdx()
		+var val: int
		+var idx: int
		+func init(val: int, idx: int): ValIdx
			do me.val :: val
			do me.idx :: idx
			ret me
		end func
		+*func cmp(t: kuin@Class): int
			var tVal: int :: (t $ ValIdx).val
			var tIdx: int :: (t $ ValIdx).idx
			if(me.val < tVal)
				ret - 1
			elif(me.val > tVal)
				ret 1
			elif(me.idx < tIdx)
				ret - 1
			elif(me.idx > tIdx)
				ret 1
			else
				ret 0
			end if
		end func
	end class
	var valIdxs: []ValIdx :: #[n]ValIdx
	for i(0, n - 1)
		do valIdxs[i] :: (#ValIdx).init(a[i], i)
	end for
	do valIdxs.sort()
	do valIdxs.reverse()
	
	var ans: list<int> :: #list<int>
	for i(n - 1, 0, -1)
		var idx: int :: valIdxs[i].idx
		if(c[idx])
			do ans.add(idx + 1)
			do uvs[idx].head()
			for j(0, ^uvs[idx] - 1)
				var v: int :: uvs[idx].get()
				if(a[idx] < a[v])
					do c[v] :: !c[v]
				end if
				do uvs[idx].next()
			end for
		end if
	end for
	do cui@print("\{^ans}\n")
	do ans.head()
	for i(0, ^ans - 1)
		do cui@print("\{ans.get()}\n")
		do ans.next()
	end for
end func
0