結果

問題 No.1890 Many Sequences Sum Queries
ユーザー 👑 tatt61880tatt61880
提出日時 2023-03-14 12:38:13
言語 Kuin
(KuinC++ v.2021.9.17)
結果
AC  
実行時間 129 ms / 2,000 ms
コード長 1,332 bytes
コンパイル時間 2,218 ms
コンパイル使用メモリ 145,952 KB
実行使用メモリ 4,504 KB
最終ジャッジ日時 2023-10-18 11:36:04
合計ジャッジ時間 4,172 ms
ジャッジサーバーID
（参考情報） 		judge14 / judge15
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 129 ms
4,504 KB
testcase_01 AC 99 ms
4,348 KB
testcase_02 AC 2 ms
4,348 KB
testcase_03 AC 1 ms
4,348 KB
testcase_04 AC 2 ms
4,348 KB
testcase_05 AC 2 ms
4,348 KB
testcase_06 AC 2 ms
4,348 KB
testcase_07 AC 50 ms
4,348 KB
testcase_08 AC 11 ms
4,348 KB
testcase_09 AC 14 ms
4,348 KB
testcase_10 AC 88 ms
4,348 KB
testcase_11 AC 43 ms
4,348 KB
testcase_12 AC 90 ms
4,348 KB
testcase_13 AC 95 ms
4,348 KB
testcase_14 AC 50 ms
4,348 KB
testcase_15 AC 1 ms
4,348 KB
testcase_16 AC 2 ms
4,348 KB
testcase_17 AC 2 ms
4,348 KB
testcase_18 AC 2 ms
4,348 KB
testcase_19 AC 2 ms
4,348 KB
testcase_20 AC 2 ms
4,348 KB
testcase_21 AC 2 ms
4,348 KB
testcase_22 AC 2 ms
4,348 KB
testcase_23 AC 1 ms
4,348 KB
testcase_24 AC 2 ms
4,348 KB
testcase_25 AC 2 ms
4,348 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

func main()
	var n: int :: cui@inputInt()
	var q: int :: cui@inputInt()
	
	var a: []int :: #[n]int
	for i(0, n - 1)
		do a[i] :: cui@inputInt()
	end for
	
	; Pの累積和
	var cumP: []int :: #[a.max() + 1]int
	do cumP[0] :: 0
	for i(1, a.max())
		do cumP[i] :: cumP[i - 1] + i
	end for
	
	; idxの累積和
	var cumIdx: []int :: #[n + 1]int
	do cumIdx[0] :: 0
	for i(1, n)
		do cumIdx[i] :: cumIdx[i - 1] + a[i - 1]
	end for
	
	; valの累積和
	var cumVal: []int :: #[n + 1]int
	do cumVal[0] :: 0
	for i(1, n)
		do cumVal[i] :: cumVal[i - 1] + cumP[a[i - 1]]
	end for
	
	for loop(1, q)
		var s: int :: cui@inputInt()
		if(s > cumVal[n])
			do cui@print("-1\n")
			skip loop
		end if
		
		var idx: int
		block
			; valの累積和がsを超えなければok
			var ok: int :: 0
			var ng: int :: n + 1
			while((ok - ng).abs() > 1)
				var mid: int :: (ok + ng) / 2
				if(cumVal[mid] <= s)
					do ok :: mid
				else
					do ng :: mid
				end if
			end while
			do idx :: ok
		end block
		
		do s :- cumVal[idx]
		var ans: int :: cumIdx[idx]
		if(idx <> n)
			var ok: int :: a[idx]
			var ng: int :: -1
			while((ok - ng).abs() > 1)
				var mid: int :: (ok + ng) / 2
				if(cumP[mid] >= s)
					do ok :: mid
				else
					do ng :: mid
				end if
			end while
			do ans :+ ok
		end if
		do cui@print("\{ans}\n")
	end for
end func
0