CF 13E Holes（分塊處理）

acm_cxlove發表於2013-05-08

轉載請註明出處，謝謝http://blog.csdn.net/ACM_cxlove?viewmode=contents    by---cxlove

開始以為是用並查集去維護，因為某些位置會形成一條鏈，記錄根的位置以及鏈的長就可以解決。
但是修改操作一直無法實現，並不是簡單的刪除再合併。
解法是分塊處理。一個位置進行修改，並不需要修改前面的所有位置
我們只記錄某個位置在當前這個塊的層數，以及最後一個位置
同樣還要記錄到達下一塊的位置
修改的時候，只需要修改當前塊的所有位置
查詢的時候，根據next，遍歷一下所有塊
這一塊弱爆了。。。。
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <cmath>
#define lowbit(x) ((x)&(-(x)))
using namespace std;
const int N = 100005;
int n,m,a[N];
int cnt[N],end[N],next[N]; 
int BLOCK;   //每個塊數目 
void add(int i,int j){
    if(j>n){
        cnt[i]=1;
        end[i]=i;
        next[i]=n+1;
    }
    else{
        if(i/BLOCK==j/BLOCK){
            cnt[i]=cnt[j]+1; 
            end[i]=end[j];
            next[i]=next[j];
        }
        else{
            next[i]=j;
            end[i]=i;
            cnt[i]=1;
        }
    }
}
void ask(int u){
    int c=cnt[u],e=end[u];
    while(true){
        u=next[u];
        if(u>n) break;
        c+=cnt[u];
        e=end[u];
    }
    printf("%d %d\n",e,c);
}
int main(){
    scanf("%d%d",&n,&m);
    BLOCK=ceil(sqrt(n*1.0));
    for(int i=1;i<=n;i++)
        scanf("%d",&a[i]);
    for(int i=n;i>=1;i--)
        add(i,i+a[i]);
    while(m--){
        int k,u,v;
        scanf("%d",&k);
        if(k){
            scanf("%d",&u);
            ask(u);
        }
        else{
            scanf("%d%d",&u,&v);
            add(u,u+v);
            for(int i=u-1;i>=u/BLOCK*BLOCK;i--)
                add(i,i+a[i]);
            a[u]=v;
        }
    }
    return 0;
}


相關文章