loj #6279. 數列分塊入門 3

#6279. 數列分塊入門 3

題目描述

給出一個長爲 nnn 的數列，以及 nnn 個操做，操做涉及區間加法，詢問區間內小於某個值 xxx 的前驅（比其小的最大元素）。

輸入格式

第一行輸入一個數字 nnn。

第二行輸入 nnn 個數字，第 i 個數字爲 aia_ia​i​​，以空格隔開。

接下來輸入 nnn 行詢問，每行輸入四個數字 opt\mathrm{opt}opt、lll、rrr、ccc，以空格隔開。

若 opt=0\mathrm{opt} = 0opt=0，表示將位於 [l,r][l, r][l,r] 的之間的數字都加 ccc。

若 opt=1\mathrm{opt} = 1opt=1，表示詢問 [l,r][l, r][l,r] 中 ccc 的前驅的值（不存在則輸出 −1-1−1）。

輸出格式

對於每次詢問，輸出一行一個數字表示答案。

樣例

樣例輸入

4
1 2 2 3
0 1 3 1
1 1 4 4
0 1 2 2
1 1 2 4

樣例輸出

3
-1

數據範圍與提示

對於 100% 100\%100% 的數據，1≤n≤100000,−231≤others 1 \leq n \leq 100000, -2^{31} \leq \mathrm{others}1≤n≤100000,−2​31​​≤others、ans≤231−1 \mathrm{ans} \leq 2^{31}-1ans≤2​31​​−1。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define maxn 100010
using namespace std;
int a[maxn],b[maxn],n;
int find(int l,int r,int c){
    int res=-1;
    while(l<=r){
        int mid=(l+r)>>1;
        if(b[mid]<c)res=b[mid],l=mid+1;
        else r=mid-1;
    }
    return res;
}
int main(){
    scanf("%d",&n);
    for(int i=1;i<=n;i++)scanf("%d",&a[i]);
    int op,l,r,c;
    for(int i=1;i<=n;i++){
        scanf("%d%d%d%d",&op,&l,&r,&c);
        if(op==0)
            for(int j=l;j<=r;j++)
                a[j]+=c;
        else {
            for(int j=l;j<=r;j++)
                b[j]=a[j];
            sort(b+l,b+r+1);
            printf("%d\n",find(l,r,c));
        }
    }
    return 0;
}

30分 暴力

#include<iostream>
#include<cstdio>
#include<cstring>
#include<set>
#define maxn 100010
using namespace std;
int n,block,a[maxn],b[maxn],pos[maxn];
set<int>s[110];
void modify(int l,int r,int c){
    for(int i=l;i<=min(pos[l]*block,r);i++){
        s[pos[i]].erase(a[i]);
        a[i]+=c;
        s[pos[i]].insert(a[i]);
    }
    if(pos[l]!=pos[r]){
        for(int i=(pos[r]-1)*block+1;i<=r;i++){
            s[pos[i]].erase(a[i]);
            a[i]+=c;
            s[pos[i]].insert(a[i]);
        }
    }
    for(int i=pos[l]+1;i<=pos[r]-1;i++)b[i]+=c;
}
void query(int l,int r,int c){
    int ans=-1;
    for(int i=l;i<=min(pos[l]*block,r);i++)
        if(a[i]+b[pos[i]]<c)ans=max(ans,a[i]+b[pos[i]]);
    if(pos[l]!=pos[r])
        for(int i=(pos[r]-1)*block+1;i<=r;i++)
        if(a[i]+b[pos[i]]<c)ans=max(ans,a[i]+b[pos[i]]);
    for(int i=pos[l]+1;i<=pos[r]-1;i++){
        int x=c-b[i];
        set<int>::iterator it=s[i].lower_bound(x);
        if(it==s[i].begin())continue;
        it--;
        ans=max(ans,*it+b[i]);
    }
    printf("%d\n",ans);
}
int main(){
    scanf("%d",&n);
    for(int i=1;i<=n;i++)scanf("%d",&a[i]);
    block=1000;
    for(int i=1;i<=n;i++){
        pos[i]=(i-1)/block+1;
        s[pos[i]].insert(a[i]);
    }
    for(int i=1;i<=n;i++){
        int op,l,r,c;
        scanf("%d%d%d%d",&op,&l,&r,&c);
        if(op==0)modify(l,r,c);
        else query(l,r,c);
    }
    return 0;
}

100分 分塊+set