結果

問題 No.1890 Many Sequences Sum Queries
ユーザー tatt61880tatt61880
提出日時 2023-03-14 12:39:22
言語 Kuin
(KuinC++ v.2021.9.17)
結果
AC  
実行時間 125 ms / 2,000 ms
コード長 1,371 bytes
コンパイル時間 2,191 ms
コンパイル使用メモリ 146,652 KB
実行使用メモリ 6,944 KB
最終ジャッジ日時 2024-09-18 07:59:42
合計ジャッジ時間 3,745 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 125 ms
6,812 KB
testcase_01 AC 89 ms
6,940 KB
testcase_02 AC 2 ms
6,940 KB
testcase_03 AC 2 ms
6,940 KB
testcase_04 AC 2 ms
6,944 KB
testcase_05 AC 2 ms
6,940 KB
testcase_06 AC 2 ms
6,940 KB
testcase_07 AC 47 ms
6,944 KB
testcase_08 AC 9 ms
6,944 KB
testcase_09 AC 13 ms
6,940 KB
testcase_10 AC 79 ms
6,940 KB
testcase_11 AC 39 ms
6,944 KB
testcase_12 AC 81 ms
6,940 KB
testcase_13 AC 86 ms
6,940 KB
testcase_14 AC 46 ms
6,940 KB
testcase_15 AC 1 ms
6,944 KB
testcase_16 AC 1 ms
6,940 KB
testcase_17 AC 2 ms
6,940 KB
testcase_18 AC 2 ms
6,940 KB
testcase_19 AC 2 ms
6,940 KB
testcase_20 AC 1 ms
6,940 KB
testcase_21 AC 1 ms
6,940 KB
testcase_22 AC 1 ms
6,940 KB
testcase_23 AC 2 ms
6,944 KB
testcase_24 AC 1 ms
6,940 KB
testcase_25 AC 1 ms
6,940 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)
			; 累積和がs以上の場合にok
			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