結果

問題 No.1890 Many Sequences Sum Queries
ユーザー trineutron
提出日時 2022-04-04 21:14:20
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 34 ms / 2,000 ms
コード長 584 bytes
コンパイル時間 1,916 ms
コンパイル使用メモリ 196,792 KB
最終ジャッジ日時 2025-01-28 15:03:38
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 25
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:29:34: warning: format ‘%d’ expects argument of type ‘int’, but argument 2 has type ‘__gnu_cxx::__normal_iterator<long int*, std::vector<long int> >::difference_type’ {aka ‘long int’} [-Wformat=]
   29 |                         printf("%d\n", it - c.begin());
      |                                 ~^     ~~~~~~~~~~~~~~
      |                                  |        |
      |                                  int      __gnu_cxx::__normal_iterator<long int*, std::vector<long int> >::difference_type {aka long int}
      |                                 %ld
main.cpp:5:14: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
    5 |         scanf("%d%d", &n, &q);
      |         ~~~~~^~~~~~~~~~~~~~~~
main.cpp:8:22: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
    8 |                 scanf("%d", &a[i]);
      |                 ~~~~~^~~~~~~~~~~~~
main.cpp:24:22: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   24 |                 scanf("%d", &s);
      |                 ~~~~~^~~~~~~~~~

ソースコード

diff # 

#include <bits/stdc++.h>

int main() {
	int n, q, len = 0;
	scanf("%d%d", &n, &q);
	std::vector<int> a(n);
	for (int i = 0; i < n; i++) {
		scanf("%d", &a[i]);
		len += a[i];
	}
	std::vector<int64_t> c(len + 1);
	int idx = 1;
	for (int i = 0; i < n; i++) {
		for (int j = 0; j < a[i]; j++) {
			c[idx] = j + 1;
			idx++;
		}
	}
	for (int i = 0; i < len; i++) {
		c[i + 1] += c[i];
	}
	for (int i = 0; i < q; i++) {
		int s;
		scanf("%d", &s);
		auto it = lower_bound(c.begin(), c.end(), s);
		if (it == c.end()) {
			puts("-1");
		} else {
			printf("%d\n", it - c.begin());
		}
	}
}
0