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

在这里插入图片描述
在这里插入图片描述
思路:考虑差分,type为1的时候就是每个数加1,为2时就是每个数加x,为3时就是加x的平方,这个题难就难在如何实现差分数组,有个大佬的博客很好,推荐不会的可以去学一下,大佬博客于是答案就是ans=x^2+x+1,其中x是从当前位与pos位的间距,分别用三个数组维护多项式的三个部分。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=1e5+1;
const int mod=1e9+7;
ll a[maxn],b[maxn],c[maxn];
void add(ll *d,int n)
{
    for(int i=1;i<=n;++i)
    d[i]=(d[i]+d[i-1])%mod;
}
int  main()
{
    int T,n,m,type,pos;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d %d",&n,&m);
        for(int i=1;i<=n;++i) a[i]=b[i]=c[i]=0;
        while(m--)
        {
            scanf("%d%d",&type,&pos);
            if(type==1) a[pos]++;
            else if(type==2) b[pos]++;
            else c[pos]++,c[pos+1]++;
        }
        add(a,n);
        for(int i=1;i<=2;++i) add(b,n);
        for(int i=1;i<=3;++i) add(c,n);
        for(int i=1;i<=n;++i) printf("%lld%s",(a[i]+b[i]+c[i])%mod,(i==n)?"\n":" ");
    }
}
发布了70 篇原创文章 · 获赞 0 · 访问量 2448

猜你喜欢

转载自blog.csdn.net/qq_42479630/article/details/104138191