結果

問題 No.1890 Many Sequences Sum Queries
ユーザー tatt61880
提出日時 2023-03-14 12:27:38
言語 Kuin
(KuinC++ v.2021.9.17)
結果
RE  
実行時間 -
コード長 1,269 bytes
コンパイル時間 2,030 ms
コンパイル使用メモリ 146,512 KB
実行使用メモリ 6,944 KB
最終ジャッジ日時 2024-09-18 07:59:10
合計ジャッジ時間 6,975 ms
ジャッジサーバーID
(参考情報)
judge3 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample RE * 1
other RE * 25
権限があれば一括ダウンロードができます

ソースコード

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