結果

問題 No.1890 Many Sequences Sum Queries
ユーザー shisidor
提出日時 2023-04-01 19:27:35
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 185 ms / 2,000 ms
コード長 701 bytes
コンパイル時間 763 ms
コンパイル使用メモリ 72,492 KB
実行使用メモリ 6,944 KB
最終ジャッジ日時 2024-09-24 11:12:54
合計ジャッジ時間 3,108 ms
ジャッジサーバーID
(参考情報)
judge2 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 1
other AC * 25
権限があれば一括ダウンロードができます

ソースコード

diff #

#include<iostream>
#include<vector>
using namespace std;

int main() {
    int n, q;
    cin >> n >> q;
    vector<int> a(n);
    for (int i = 0; i < n; i++) cin >> a[i];
    vector<long long> b;
    for (int i = 0; i < n; i++) {
        for (int j = 1; j <= a[i]; j++) {
            b.push_back(j);
        }
    }
    for (int i = 1; i < (int)b.size(); i++) {
        b[i] += b[i - 1];
    }
    for (int i = 0; i < q; i++) {
        long long s;
        cin >> s;
        auto itr = lower_bound(b.begin(), b.end(), s);
        if (itr == b.end()) cout << -1 << endl;
        else {
            int idx = distance(b.begin(), itr);
            cout << idx + 1 << endl;
        }
    }
    return 0;
}
0