UVA12003 Array Transformer(数组分块)

UVA12003 Array Transformer(数组分块)

题解:(分块思想)

1.把这个数组分成 s q r t ( n ) sqrt(n) sqrt(n)个块,对于每个块对他进行从小到大排序。
2.查询L-R的所在块,第一个块和最后一个块进行暴力查找,中间的块二分查找,把所有的数量累加起来即可。
3.修改直接把原数组的值给修改好,然后对于这个包含这个位置的块给全部从新赋一遍对应的原数组的值,然后对当前快进行sort即可。

/*Keep on going Never give up*/
#pragma GCC optimize(3,"Ofast","inline")
#include<bits/stdc++.h>
#define int long long
#define endl '\n'

using namespace std;
const int maxn=3e5+10;
const int mod=1e9+7;

vector<int> v[600];
int sz;
int a[maxn];

int query(int l,int r,int val){
    int lft=l/sz,rht=r/sz;
    int res=0;
    if(lft==rht){
        for(int i=l;i<=r;i++){
            if(a[i]<val) res++;
        }
    }
    else{
        for(int i=l;i<(l/sz+1)*sz;i++){  //第一块
            if(a[i]<val) res++;
        }
        for(int i=(r/sz)*sz;i<=r;i++){   //最后一块
            if(a[i]<val) res++;
        }
        for(int i=lft+1;i<rht;i++){   //中间块
            int t=lower_bound(v[i].begin(),v[i].end(),val)-v[i].begin();
            res+=t;
        }
    }
    return res;
}
int n,m,u;
void modify(int pos){
    int s=pos/sz;
    for(int i=0;i<sz;i++){
        v[s][i]=a[s*sz+i];
        if(s*sz+i==n-1) break;
    }
    sort(v[s].begin(),v[s].end());
}
signed main() {
    ios::sync_with_stdio(false);
    cin.tie(0);
    cin>>n>>m>>u;
    sz=sqrt(n);
    for(int i=0;i<n;i++){
        cin>>a[i];
        v[i/sz].push_back(a[i]);
    }
    for(int i=0;i<=sz;i++){
        sort(v[i].begin(),v[i].end());
    }
    while(m--){
        int l,r,val,p;
        cin>>l>>r>>val>>p;
        int xx=query(l-1,r-1,val);
        a[p-1]=(u*xx)/(r-l+1);
        modify(p-1);
    }
    for(int i=0;i<n;i++){
        cout<<a[i]<<endl;
    }

}

上一篇:打卡学习 1-28


下一篇:C语言数组(下)