P3863 序列

题目通道(序列 - 洛谷)

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=300010,B=333;
int n,q,ans[N],L[N],R[N],id[N];
ll tag[N];//每个块的区间加懒标记
struct A{
    int pos;//pos是原本的位置，存下才能判断它需不需要修改/查询(每个块内部会打乱顺序的）
    ll v;//V是值
    bool operator<(const A &t){
        if(v!=t.v) return v<t.v;
        else return pos<t.pos;
    }
}a[N];
struct node{int t,pos,x;};//t=1:修改，t=2:查询  pos是修改/查询的时间维度上位置，X是增加的值/查询的值
vector<node> Do[N];//（序列维度）桶排序
inline void modify(int l,int x)
{
    for(int i=L[id[l]];i<=R[id[l]];i++)//块内
        if(a[i].pos>=l) //属于[l,q]
            a[i].v+=x;
    sort(a+L[id[l]],a+R[id[l]]+1);//注意这个序列是时间，所以排序是可以的
    for(int i=id[l]+1;i<=(q-1)/B+1;i++) tag[i]+=x;
}
inline int query(int r,int x)
{
    int ans=0;
    for(int i=L[id[r]];i<=R[id[r]];i++)
        if(a[i].pos<=r&&a[i].v+tag[id[r]]>=x)
            ans++;
    for(int i=1;i<id[r];i++) 
        ans+=R[i]-(lower_bound(a+L[i],a+R[i]+1,A({0,x-tag[i]}))-a)+1;
    return ans;
}
int main()
{
    memset(ans,-1,sizeof ans);//标记一下没有询问，否则输出的时候不知道0是这里根本没有询问还是答案是0
    scanf("%d%d",&n,&q); 
    q++;//把所有修改询问向后移一个，把Q=1当作初始值
    for(int i=1,a;i<=n;i++)
    {
        scanf("%d",&a);
        Do[i].push_back({1,1,a});
        Do[i+1].push_back({1,1,-a});//这两行相当于只改[i,i]
    }
    for(int i=2,op,l,r,x,p,y;i<=q;i++)
    {
        scanf("%d",&op);
        if(op==1) 
        {
            scanf("%d%d%d",&l,&r,&x);
            Do[l].push_back({1,i,x});
            Do[r+1].push_back({1,i,-x});//这两行相当于改[l,r]
        }
        else scanf("%d%d",&p,&y),Do[p].push_back({2,i,y});
    }
    for(int i=1;i<=q;i++) a[i]={i,0},id[i]=(i-1)/B+1;
    for(int i=1;i<=(q-1)/B+1;i++) L[i]=B*(i-1)+1,R[i]=min(B*i,q);
    for(int i=1;i<=n;i++)
        for(node t:Do[i])//操作序列上的每个位置
        {
            if(t.t==1) modify(t.pos,t.x);//修改[t.pos,q]
            else ans[t.pos]=query(t.pos-1,t.x);//查询[1,t.pos-1]
        }
    for(int i=2;i<=q;i++) if(~ans[i]) printf("%d\n",ans[i]);
    return 0;
}