牛客网暑期ACM多校训练营(第十场) Rikka with Prefix Sum

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/tianwei0822/article/details/81914694

题目:点击打开链接

题意:给一个数组a,一开始的值全为0。一共有三个操作: 
1. 对区间[L,R]的每个数都加上w。 
2. 将数组a用其前缀和数组代替。 
3. 将询问区间[L,R]的区间和。

分析:好题,加深了对差分数组的理解。

操作有两种,1操作是给l-r区间内的数都加w,2操作是让这个数列变为它的前缀和序列,我们知道,2操作之后得到的新的序列差分之后就是操作前的序列,所以如果只有2操作的话,就是给你一个差分了很多次之后的序列求原序列,但是它还有1操作,1操作对于差分的级别没有变化,但我们知道在原来的序列的l-r区间+w,其实就是在它的差分序列l处+w,r+1处-w,那么现在的问题就在于在这样的一个差分序列的表格中,如果我们在某个点+w,造成的影响是什么。

我们发现在一个点+w之后,影响的是它右下角的所有点,每个行的系数是杨辉三角的一列,所以如果在(i,j)点+w,那么(x,y)点的系数为C(x-ai+y-j-1,x-i-1)

然后3操作不超过500次,所以我们就记录下每次操作,然后对每次询问O(n)计算即可

题解:

代码:

#pragma comment(linker, "/STACK:102400000,102400000")
#include<algorithm>
#include<iostream>
#include<fstream>
#include<complex>
#include<cstdlib>
#include<cstring>
#include<cassert>
#include<iomanip>
#include<string>
#include<cstdio>
#include<bitset>
#include<vector>
#include<cctype>
#include<cmath>
#include<ctime>
#include<stack>
#include<queue>
#include<deque>
#include<list>
#include<set>
#include<map>
using namespace std;
#define pt(a) cout<<a<<endl
#define debug test
#define mst(ss,b) memset((ss),(b),sizeof(ss))
#define rep(i,a,n) for (int i=a;i<=n;i++)
#define per(i,a,n) for (int i=n-1;i>=a;i--)
#define all(x) (x).begin(),(x).end()
#define fi first
#define se second
#define SZ(x) ((int)(x).size())
#define ll long long
#define ull unsigned long long
#define pb push_back
#define mp make_pair
#define inf 0x3f3f3f3f
#define eps 1e-10
#define PI acos(-1.0)
typedef pair<int,int> PII;
const ll mod = 998244353;
const int N = 1e6+10;

ll gcd(ll p,ll q){return q==0?p:gcd(q,p%q);}
ll qp(ll a,ll b) {ll res=1;a%=mod; assert(b>=0); for(;b;b>>=1){if(b&1)res=res*a%mod;a=a*a%mod;}return res;}
int to[4][2]={{-1,0},{1,0},{0,-1},{0,1}};

ll t,n,m,a[N],f[N],inv[N],ct;

struct nd{
    ll x,y,w;
}q[N];

void init() {
    f[0]=f[1]=1;
    rep(i,2,N-1) f[i]=f[i-1]*i%mod;
    inv[0]=inv[1]=1;
    rep(i,2,N-1) inv[i]=(mod-mod/i)*inv[mod%i]%mod;
    rep(i,2,N-1) inv[i]=inv[i-1]*inv[i]%mod;
}

ll C(ll x,ll y) {
    if(y<0||y>x) return 0;
    else return f[x]*inv[y]%mod*inv[x-y]%mod;
}

ll sv(ll x,ll y) {
    ll res=0;
    rep(i,1,ct)
        if(q[i].x<=x&&q[i].y<=y)
            (res += C(x-q[i].x+y-q[i].y-1,x-q[i].x-1)*q[i].w%mod)%=mod;
    return res;
}

int main() {
    ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
    init();
    cin>>t;
    while(t--) {
        cin>>n>>m;
        ll k=0,op,l,r,w;
        ct=0;
        rep(i,1,m) {
            cin>>op;
            if(op==1) {
                cin>>l>>r>>w;
                ct++,q[ct].x=k,q[ct].y=l,q[ct].w=w;
                ct++,q[ct].x=k,q[ct].y=r+1,q[ct].w=(mod-w)%mod;
            }else if(op==2) k++;
            else {
                cin>>l>>r;
                cout<<((sv(k+2,r)-sv(k+2,l-1))%mod+mod)%mod<<endl;
            }
        }
    }
    return 0;
}

参考博客:https://www.nowcoder.com/acm/contest/148/D

猜你喜欢

转载自blog.csdn.net/tianwei0822/article/details/81914694