HDU 4638 Group(セグメントツリー|ツリー配列+オフライン処理)

19169 ワード

Group
Time Limit: 4000/2000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)Total Submission(s): 208    Accepted Submission(s): 122
Problem Description
There are n men ,every man has an ID(1..n).their ID is unique. Whose ID is i and i-1 are friends, Whose ID is i and i+1 are friends. These n men stand in line. Now we select an interval of men to make some group. K men in a group can create K*K value. The value of an interval is sum of these value of groups.
 The people of same group's id must be continuous. Now we chose an interval of men and want to know there should be how many groups so the value of interval is max.
 
 
Input
First line is T indicate the case number.
For each case first line is n, m(1<=n ,m<=100000) indicate there are n men and m query.
Then a line have n number indicate the ID of men from left to right.
Next m line each line has two number L,R(1<=L<=R<=n),mean we want to know the answer of [L,R].
 
 
Output
For every query output a number indicate there should be how many group so that the sum of value is max.
 
 
Sample Input
1 5 2 3 1 2 5 4 1 5 2 4
 
 
Sample Output
1 2
 
 
Source
2013 Multi-University Training Contest 4
 
 
Recommend
zhuyuanchen520
 
标题:区間エネルギーを見つけて、何組の連続数字列があるか 
 (転)構想:明らかにvalueを最大にするには、できるだけ連続したIDを1つのグループに分ける必要があるので、問題は1つの区間における連続したID区間の個数を求めることに転化する.左から右へ走査し,右端点をiとする問合せを順次考慮し,dp[l]を区間[l,i]の連続区間個数とし,po[i]をiが出現する位置とし,まだ出現していない場合は0とし,現在考慮している右端点をa[i]とし,まずa[i]が区間[1,i-1]のいずれかの数と1組に分けられないと仮定すると,dp[1]からdp[i-1]にすべて1を加算し,po[a]を考慮する.[i]+1]が0でないかどうかは、0でなければa[i]-1が既に前に現れていることを示し、dp[1]からdp[po[a[i]+1]まですべて1つを減らす必要がある.a[i]はa[i]+1とグループ化できるため、私たちが前に加えた1は余計である.a[i]について-1の場合も同じです.以上の操作は線分木や木の配列などで実現できます.その後、質問を右端点で小さいから大きいまで並べ替えて、オフラインで処理すればいいです.以下はコードで実現します.
線分ツリー:
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>

using namespace std;

const int N=100010;

#define L(rt) (rt<<1)
#define R(rt) (rt<<1|1)

struct Tree{
    int l,r;
    int num;    //num  (l,r)        
}tree[N<<2];

struct node{
    int l,r;
    int id;
}f[N];

int cmp(node a,node b){
    return a.r<b.r;
}

int a[N],loc[N],res[N];

void build(int L,int R,int rt){
    tree[rt].l=L;
    tree[rt].r=R;
    tree[rt].num=0;
    if(tree[rt].l==tree[rt].r)
        return ;
    int mid=(L+R)>>1;
    build(L,mid,L(rt));
    build(mid+1,R,R(rt));
}

void update(int L,int R,int val,int rt){
    if(tree[rt].l==L && tree[rt].r==R){
        tree[rt].num+=val;
        return ;
    }
    int mid=(tree[rt].l+tree[rt].r)>>1;
    if(R<=mid)
        update(L,R,val,L(rt));
    else if(L>=mid+1)
        update(L,R,val,R(rt));
    else{
        update(L,mid,val,L(rt));
        update(mid+1,R,val,R(rt));
    }
    tree[rt].num=tree[L(rt)].num+tree[R(rt)].num;
}

int query(int L,int R,int rt){
    if(tree[rt].l==L && tree[rt].r==R)
        return tree[rt].num;
    int mid=(tree[rt].l+tree[rt].r)>>1;
    if(R<=mid)
        return query(L,R,L(rt));
    else if(L>=mid+1)
        return query(L,R,R(rt));
    else{
        int a=query(L,mid,L(rt));
        int b=query(mid+1,R,R(rt));
        return a+b;
    }
}

int main(){

    //freopen("input.txt","r",stdin);

    int t,n,m;
    scanf("%d",&t);
    while(t--){
        scanf("%d%d",&n,&m);
        for(int i=1;i<=n;i++){
            scanf("%d",&a[i]);
            loc[a[i]]=i;
        }
        for(int i=1;i<=m;i++){
            scanf("%d%d",&f[i].l,&f[i].r);
            f[i].id=i;
        }
        sort(f+1,f+m+1,cmp);    //              ,             ,    
        build(1,n,1);
        int i,j=1;
        for(i=1;i<=n;i++){
            update(i,i,1,1);    //        ,    ,   i      +1 
            if(a[i]<n && loc[a[i]+1]<i)     //           a[i]    ,        
                update(loc[a[i]+1],loc[a[i]+1],-1,1);
            if(a[i]>1 && loc[a[i]-1]<i)
                update(loc[a[i]-1],loc[a[i]-1],-1,1);
            while(j<=m && f[j].r==i){
                res[f[j].id]=query(f[j].l,f[j].r,1);
                j++;
            }
        }
        for(i=1;i<=m;i++)
            printf("%d
",res[i]); } return 0; }

 
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>

using namespace std;

const int N=100010;

int n,m;
int arr[N],a[N],loc[N],res[N];

struct node{
    int l,r;
    int id;
}f[N];

int cmp(node a,node b){
    return a.r<b.r;
}

int lowbit(int x){
    return x&(-x);
}

void update(int i,int val){
    while(i<=n){
        arr[i]+=val;
        i+=lowbit(i);
    }
}

int sum(int i){
    int ans=0;
    while(i>0){
        ans+=arr[i];
        i-=lowbit(i);
    }
    return ans;
}

int main(){

    //freopen("input.txt","r",stdin);

    int t;  //        int t,n,m;            main           n(    0),      。。。。。
    scanf("%d",&t);
    while(t--){

        scanf("%d%d",&n,&m);
        for(int i=1;i<=n;i++){
            scanf("%d",&a[i]);
            loc[a[i]]=i;
        }
        for(int i=1;i<=m;i++){
            scanf("%d%d",&f[i].l,&f[i].r);
            f[i].id=i;
        }
        sort(f+1,f+m+1,cmp);    //              ,             ,    
        memset(arr,0,sizeof(arr));
        int i,j=1;

        for(i=1;i<=n;i++){
            update(i,1);    //        ,    ,   i      +1
            if(a[i]<n && loc[a[i]+1]<i)     //           a[i]    ,        
                update(loc[a[i]+1],-1);
            if(a[i]>1 && loc[a[i]-1]<i)
                update(loc[a[i]-1],-1);
            while(j<=m && f[j].r==i){
                res[f[j].id]=sum(f[j].r)-sum(f[j].l-1);
                j++;
            }
        }
        for(i=1;i<=m;i++)
            printf("%d
",res[i]); } return 0; }