結果

問題 No.1890 Many Sequences Sum Queries
ユーザー 👑 tatt61880tatt61880
提出日時 2023-03-14 12:27:38
言語 Kuin
(KuinC++ v.2021.9.17)
結果
RE  
実行時間 -
コード長 1,269 bytes
コンパイル時間 4,902 ms
コンパイル使用メモリ 146,028 KB
実行使用メモリ 4,348 KB
最終ジャッジ日時 2023-10-18 11:35:32
合計ジャッジ時間 9,561 ms
ジャッジサーバーID
（参考情報） 		judge15 / judge14
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 RE -
testcase_01 RE -
testcase_02 RE -
testcase_03 RE -
testcase_04 RE -
testcase_05 RE -
testcase_06 RE -
testcase_07 RE -
testcase_08 RE -
testcase_09 RE -
testcase_10 RE -
testcase_11 RE -
testcase_12 RE -
testcase_13 RE -
testcase_14 RE -
testcase_15 RE -
testcase_16 RE -
testcase_17 RE -
testcase_18 RE -
testcase_19 RE -
testcase_20 RE -
testcase_21 RE -
testcase_22 RE -
testcase_23 RE -
testcase_24 RE -
testcase_25 RE -
権限があれば一括ダウンロードができます

ソースコード

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 :: #[n]int
	do cumP[0] :: 0
	for i(1, a.max())
		do cumP[i] :: cumP[i - 1] + i
	end for
	
	; idxの累積和
	var cumIdx: []int :: #[n]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]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
			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 :: 0
			var ng: int :: a[idx] + 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