Codeforces #179(Div 2)C Greg and Array

題目鏈接:http://codeforces.com/contest/296/problem/C

題意: 一個序列中有n個數,有m個操作,每個操作將從a到b的數加上d,接下來有k對數,每對由a,b組成,代表把從a到b的操作都執行一次。問將k對數代表的操作全部執行完後序列中的數各是多少。


思路:初看覺得應該用線段樹區間更新實現,但是要寫兩棵樹,有些麻煩。但是對於每個操作只需要記錄這個操作一共進行了多少次,而不需要進行求最值,合併這類操作。所以線段樹是沒有必要的。用cnt數組記錄每個操作進行了多少次。對於a,b cnt[a] + 1,cnt[b+1] -1。最後將1~m區間上的數從左往右累加。那麼保證每個a,b區間上的操作都有記錄。因爲cnt[b+1] -= 1。所以保證對應位置的操作數量累積到b位置後,將被b+1處的-1抵消使之後的操作數量不受影響。然後用相同的方法處理數組中每個數進行操作後的結果。



#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
const int N = 1e5+10;
typedef long long LL;

int n,m,k;
LL a[N];
int l[N],r[N],ris[N];
LL off[N], cnt[N];


int main()
{
    while (scanf("%d%d%d",&n,&m,&k)==3)
    {
        int x,y;
        memset(cnt,0,sizeof (cnt));
        memset(off,0,sizeof (off));
        for (int i=1;i<=n;i++)
        {
            scanf("%lld",&a[i]);
        }
        for (int i=1;i<=m;i++)
        {
            scanf("%d%d%d",&l[i], &r[i], &ris[i]);
        }
        for (int i=1;i<=k;i++)
        {
            scanf("%d%d",&x,&y);
            cnt[x]++;
            cnt[y+1]--;
        }
        for (int i=2;i<=m;i++)
        {
            cnt[i] += cnt[i-1];
        }

        for (int i=1;i<=m;i++)
        {
            off[l[i]] += 1LL * ris[i] * cnt[i];
            off[r[i]+1] -= 1LL * ris[i] * cnt[i];
        }

        for (int i=2;i<=n;i++)
            off[i] += off[i-1];

        for (int i=1;i<=n;i++)
            printf("%lld ",a[i] + off[i]);
        printf("\n");

    }
    return 0;
}





發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章