数列分块入门 3

题目链接
根据数列分块入门2的思想做。可能用其他的STL比较方便，我这里还是用的vector。

#include<stdio.h>
#include<iostream>
#include<algorithm>
#include<queue>
#include<cmath>
#include<string.h>
#include<string>
#include<vector>
#include<set>
using namespace std;
typedef long long int ll;
const int N=100000+10;
#define int ll
int v[N],pos[N],M,tag[N];
int n;
vector<ll>s[N];
void uppdate(int x)
{
    s[x].clear();
    for(int i=(x-1)*M+1;i<=min(x*M,n);i++)
    {
        s[x].push_back(v[i]);
    }
    sort(s[x].begin(),s[x].end());
}
void add(int l,int r,int c)
{
    for(int i=l;i<=min(pos[l]*M,r);i++)v[i]+=c;
    uppdate(pos[l]);
    if(pos[l]!=pos[r])
    {
        for(int i=(pos[r]-1)*M+1;i<=r;i++)v[i]+=c;
        uppdate(pos[r]);
    }
    for(int i=pos[l]+1;i<=pos[r]-1;i++)
        tag[i]+=c;
}
int query(int l,int r,int c)
{
    int ans=-1;
    for(int i=l;i<=min(pos[l]*M,r);i++)
    if(v[i]+tag[pos[l]]<c)ans=max(v[i]+tag[pos[l]],ans);
    for(int i=pos[l]+1;i<=pos[r]-1;i++)
    {
        int x=c-tag[i];
        int p=lower_bound(s[i].begin(),s[i].end(),x)-s[i].begin();
        if(p!=0)
        ans=max(ans,(int)s[i][p-1]+tag[i]);
    }
    if(pos[l]!=pos[r])
    {
        for(int i=(pos[r]-1)*M+1;i<=r;i++)
        if(v[i]+tag[pos[r]]<c)ans=max(v[i]+tag[pos[r]],ans);
    }
    return ans;
}
signed main()
{
    scanf("%lld",&n);M=sqrt(n);
    for(int i=1;i<=n;i++)scanf("%lld",v+i);
    for(int i=1;i<=n;i++)
    {
        pos[i]=(i-1)/M+1;
        s[pos[i]].push_back(v[i]);
    }
    for(int i=1;i<=n;i++)sort(s[i].begin(),s[i].end());
    for(int i=1;i<=n;i++)
    {
        int l,r,f,c;
        scanf("%lld%lld%lld%lld",&f,&l,&r,&c);
        if(f==0)add(l,r,c);
        if(f==1)printf("%lld\n",query(l,r,c));
    }
    return 0;
}

附加一份黄老师set写的代码Orz

#include<map>
#include<set>
#include<cmath>
#include<stack>
#include<queue>
#include<cstdio>
#include<vector>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#define mod 998244353
#define pi acos(-1)
#define inf 0x7fffffff
#define ll long long
using namespace std;
ll read()
{
    ll x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
int n,blo;
int v[100005],bl[100005],atag[100005];
set<int>st[105];
void add(int a,int b,int c)
{
    for(int i=a;i<=min(bl[a]*blo,b);i++)
    {
        st[bl[a]].erase(v[i]);
        v[i]+=c;
        st[bl[a]].insert(v[i]);
    }
    if(bl[a]!=bl[b])
    {
        for(int i=(bl[b]-1)*blo+1;i<=b;i++)
        {
            st[bl[b]].erase(v[i]);
            v[i]+=c;
            st[bl[b]].insert(v[i]);
        }
    }
    for(int i=bl[a]+1;i<=bl[b]-1;i++)
        atag[i]+=c;
}
int query(int a,int b,int c)
{
    int ans=-1;
    for(int i=a;i<=min(bl[a]*blo,b);i++)
    {
        int val=v[i]+atag[bl[a]];
        if(val<c)ans=max(val,ans);
    }
    if(bl[a]!=bl[b])        
        for(int i=(bl[b]-1)*blo+1;i<=b;i++)        
        {
            int val=v[i]+atag[bl[b]];
            if(val<c)ans=max(val,ans);
        }
    for(int i=bl[a]+1;i<=bl[b]-1;i++)
    {
        int x=c-atag[i];
        set<int>::iterator it=st[i].lower_bound(x);
        if(it==st[i].begin())continue;
        --it;
        ans=max(ans,*it+atag[i]);
    }
    return ans;
}
int main()
{
    n=read();blo=1000;
    for(int i=1;i<=n;i++)v[i]=read();
    for(int i=1;i<=n;i++)
    {
        bl[i]=(i-1)/blo+1;
        st[bl[i]].insert(v[i]);
    }
    for(int i=1;i<=n;i++)
    {
        int f=read(),a=read(),b=read(),c=read();
        if(f==0)add(a,b,c);
        if(f==1)printf("%d\n",query(a,b,c));
    }
    return 0;
}