HDU 3333-Turing Tree(BIT好题)

题意:

给你n个数的序列a,q个询问,每个询问给l,r,求在下标i在[l,r]的区间内不含重复数的和

分析:

这类题目觉得很好,很练思维,觉得不太好做。

用BIT维护和,我们可以从前向后扫一遍序列,当前数在前面已经出现过了,前一个该数的位置上就要减去这个数才保证正确(不重复),但查询区间是随机的,肯定相互影响,则我们用离线处理,先把所有查询区间按右边界升序排列,这样就不会影响了。

#include <map>
#include <set>
#include <list>
#include <cmath>
#include <queue>
#include <stack>
#include <cstdio>
#include <vector>
#include <string>
#include <cctype>
#include <complex>
#include <cassert>
#include <utility>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <algorithm>
using namespace std;
typedef pair<int,int> PII;
typedef long long ll;
#define lson l,m,rt<<1
#define pi acos(-1.0)
#define rson m+1,r,rt<<11
#define All 1,N,1
#define N 50010
#define read freopen("in.txt", "r", stdin)
const ll  INFll = 0x3f3f3f3f3f3f3f3fLL;
const int INF= 0x7ffffff;
const int mod =  1000000007;
ll bit[N],a[N],tmp[N],total[200010];
//bef前一个相同数的位置
int bef[N],n,m;
struct node{
int l,r,id;
}q[200010];
bool cmp(node x,node y){
    return x.r<y.r;
}
void add(int x,ll d){
    while(x<=n){
        bit[x]+=d;
        x+=(x&(-x));
    }
}
ll sum(int x){
    ll num=0;
    while(x>0){
        num+=bit[x];
        x-=(x&(-x));
    }
    return num;
}
void solve(){
    sort(tmp+1,tmp+1+n);
    sort(q,q+m,cmp);
    memset(bef,0,sizeof(bef));
    int j=0;
    for(int i=1;i<=n;++i){
        int pos=lower_bound(tmp+1,tmp+n+1,a[i])-tmp;
        if(!bef[pos]){
            add(i,a[i]);
            bef[pos]=i;
        }
        else{
            add(bef[pos],-a[i]);
            add(i,a[i]);
            bef[pos]=i;
        }
        while(q[j].r==i&&j<m){
            total[q[j].id]=sum(q[j].r)-sum(q[j].l-1);
            j++;
        }
    }
    for(int i=0;i<m;++i)
        printf("%I64d\n",total[i]);
}
int main()
{
    int t;
    scanf("%d",&t);
    while(t--){
        memset(bit,0,sizeof(bit));
        scanf("%d",&n);
        for(int i=1;i<=n;++i){
            scanf("%I64d",&a[i]);
            tmp[i]=a[i];
        }
        scanf("%d",&m);
        for(int i=0;i<m;++i){
            scanf("%d%d",&q[i].l,&q[i].r);
            q[i].id=i;
        }
        solve();
    }
return 0;
}

 

posted on 2015-08-16 20:32  积跬步、至千里  阅读(117)  评论(0编辑  收藏  举报

导航