牛客练习赛48 C.小w的糖果(差分)

题目

n(n<=1e5)个人，m(m<=1e5)种操作，

操作分三种，

①[pos,n]从左到右第i个点+1

②[pos,n]从左到右第i个点+i

③[pos,n]从左到右第i个点+i的平方

询问，所有操作完之后，每个点的值

思路来源

https://ac.nowcoder.com/acm/contest/view-submission?submissionId=40787848

题解

区间+1的直接维护差分数组，单点修改，做一次前缀和

区间+1 +2 +3的，做一次差分就变成区间+1的，再做一次差分，所以维护二次差分数组，做两次前缀和

区间+1 +4 +9的，做一次差分变成+1+3+5，做两次差分变成+1+2+2+2，做三次差分变成+1+1，涉及两个点

所以维护三次差分数组，做三次前缀和

代码

#include
using namespace std;
const int mod=1e9+7;
typedef long long ll;
const int N=1e5+10;
int t,n,m;
ll a[N],b[N],c[N];
int op,pos;
void pre(ll p[],int len)
{
    for(int i=1;i<=len;++i)
    p[i]=(p[i]+p[i-1])%mod;
}
int main()
{
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d",&n,&m);
        for(int i=1;i<=n;++i)
        a[i]=b[i]=c[i]=0;
        for(int i=1;i<=m;++i)
        {
            scanf("%d%d",&op,&pos);
            if(op==1)a[pos]++;
            else if(op==2)b[pos]++;
            else if(op==3)c[pos]++,c[pos+1]++;
        }
        pre(a,n);
        pre(b,n);pre(b,n);
        pre(c,n);pre(c,n);pre(c,n);
        for(int i=1;i<=n;++i)
            printf("%lld%c",(a[i]+b[i]+c[i])%mod,i==n?'\n':' ');
    }
    return 0;
}