結果

問題 No.2101 [Cherry Alpha N] ずっとこの数列だったらいいのに
ユーザー みここ
提出日時 2022-12-12 06:02:37
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 989 ms / 6,000 ms
コード長 1,709 bytes
コンパイル時間 1,329 ms
コンパイル使用メモリ 100,228 KB
最終ジャッジ日時 2025-02-09 10:03:39
ジャッジサーバーID
(参考情報)
judge4 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 42
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <atcoder/segtree>
#include <iostream>
using namespace std;
using namespace atcoder;
typedef long long ll;
typedef pair<int, int> P;

using S = ll;

S op(S a, S b)
{
    return a + b;
}

S e()
{
    return 0;
}

struct query
{
    int d, l, r;
    int id;

    bool operator<(const query &q) const
    {
        return (d != q.d ? d < q.d : id < q.id);
    }
};

ll a[200005];
int t[200005];
P p[400005];
query qr[200005];
ll ans[200005];

int main()
{
    int n;
    cin >> n;
    for (int i = 0; i < n; i++)
    {
        cin >> a[i] >> t[i];
        p[i] = P(t[i] - 1, i);
        p[i + n] = P(t[i] + a[i] - 1, i + n);
    }
    sort(p, p + n * 2);
    segtree<S, op, e> asum(n), start(n), dcount(n);
    for (int i = 0; i < n; i++)
    {
        asum.set(i, a[i]);
        start.set(i, 0);
        dcount.set(i, 0);
    }
    int q;
    cin >> q;
    for (int i = 0; i < q; i++)
    {
        cin >> qr[i].d >> qr[i].l >> qr[i].r;
        qr[i].l--;
        qr[i].id = i;
    }
    sort(qr, qr + q);
    int j = 0;
    for (int i = 0; i < q; i++)
    {
        while (j < n * 2 && p[j].first < qr[i].d)
        {
            if (p[j].second < n)
            {
                start.set(p[j].second, p[j].first);
                dcount.set(p[j].second, 1);
            }
            else
            {
                asum.set(p[j].second - n, 0);
                start.set(p[j].second - n, 0);
                dcount.set(p[j].second - n, 0);
            }
            j++;
        }
        int l = qr[i].l, r = qr[i].r;
        ans[qr[i].id] = asum.prod(l, r) - (qr[i].d * dcount.prod(l, r) - start.prod(l, r));
    }
    for (int i = 0; i < q; i++)
    {
        cout << ans[i] << endl;
    }
}
0