HDU - 3333 Turing Tree 莫队 || 离线树状数组

HDU - 3333 Turing Tree

题意：求给定区间内的不重复的数字的和，如1 1 1 3 4 ，区间1到2就是1，区间1到5就是8

莫队

#include<iostream>
#include<algorithm>
#include<cstdlib>
#include<sstream>
#include<cstring>
#include<bitset>
#include<cstdio>
#include<string>
#include<deque>
#include<stack>
#include<cmath>
#include<queue>
#include<set>
#include<map>
using namespace std;
typedef long long ll;
const int maxn = 1e5+10;
struct node
{
    int l,r,id;

}q[maxn];
ll a[maxn],b[maxn];
int n,m,block;
ll cnt[maxn],res[maxn],ans;

bool cmp(node a,node b){//按照奇偶性排序
    if( (a.l/block)==(b.l/block) ){//当左端点位于同一个块时
        if( (a.l/block)%2 )//左端点的块序号为奇数时
            return a.r<b.r;//按照从小到大排
        else//左端点的块序号为偶数时
            return a.r>b.r;//按照从大到小排
    }
    else//当左端点不位于同一个块时
        return a.l<b.l;//按照块的位置进行排序
    //return (a.l/block)^(b.l/block) ? a.l<b.l : ( ((a.l/block)&1)?a.r<b.r:a.r>b.r );
}
void add(int x)
{
    cnt[x]++;
    if(cnt[x]==1)
        ans += b[x];
}
void del(int x)
{
    cnt[x]--;
    if(cnt[x]==0)
        ans -= b[x];
}
int main()
{
    int T;
    int l,r;
    scanf("%d",&T);
    while(T--)
    {
        memset(cnt,0,sizeof(cnt));
        scanf("%d",&n);
        block = sqrt(1.0*n);
        for(int i=1;i<=n;i++)
        {
            scanf("%lld",&a[i]);
            b[i] = a[i];
        }
        sort(b+1,b+n+1);
        int tot = unique(b+1,b+n+1)-b-1;
        for(int i=1;i<=n;i++)
            a[i] = lower_bound(b+1,b+tot+1,a[i])-b;
        scanf("%d",&m);
        for(int i=1;i<=m;i++)
        {
            scanf("%d%d",&q[i].l,&q[i].r);
            q[i].id = i;
        }

        sort(q+1,q+m+1,cmp);
        l = q[1].l , r = q[1].l-1;
        ans = 0;
        for(int i=1;i<=m;i++)
        {
            while(l>q[i].l)  add(a[--l]);
            while(r<q[i].r)  add(a[++r]);
            while(l<q[i].l)  del(a[l++]);
            while(r>q[i].r)  del(a[r--]);
            res[q[i].id] = ans;
        }
        for(int i=1;i<=m;i++)
            printf("%lld\n",res[i]);
    }
    return 0;
}

离线树状数组

思路：

将询问按右端点从小到大排序，在添加位置时，如果该位置等于某一右端点就查询。如果 i 位置对应的值 a[i] 以前出现过，就在 a[i] 以前出现的位置（pre[i]）减掉 a[i] ，然后在新位置加上 a[i]。

 网上神犇的一句话，对于要查询的区间，它的右端点固定后，那么重复的数字便是右面开始最后一次出现的

#include<iostream>
#include<algorithm>
#include<cstdlib>
#include<sstream>
#include<cstring>
#include<bitset>
#include<cstdio>
#include<string>
#include<deque>
#include<stack>
#include<cmath>
#include<queue>
#include<set>
#include<map>
using namespace std;
typedef long long ll;
const int maxn = 1e5+10;
struct node
{
    int l,r,id;

}q[maxn];
ll a[maxn],b[maxn],pre[maxn];  //pre[i]  记录 i 上一次出现的位置
ll c[maxn],res[maxn];
int n,m;
bool cmp(node a,node b)
{
    return a.r<b.r;
}
int lowbit(int x)
{
    return x&(-x);
}
void add(int x,ll v)
{
    while(x<=n)
    {
        c[x] += v;
        x += lowbit(x);
    }
}
ll ask(int x)
{
    ll ans = 0;
    while(x)
    {
        ans += c[x];
        x -= lowbit(x);
    }
    return ans;
}

int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        memset(c,0,sizeof(c));
        memset(pre,0,sizeof(pre));
        scanf("%d",&n);
        for(int i=1;i<=n;i++)
        {
            scanf("%lld",&a[i]);
            b[i] = a[i];
        }
        sort(b+1,b+n+1);
        int tot = unique(b+1,b+1+n)-b-1;
        scanf("%d",&m);
        for(int i=1;i<=m;i++)
        {
            scanf("%d%d",&q[i].l,&q[i].r);
            q[i].id = i;
        }
        sort(q+1,q+m+1,cmp);
        for(int i=1,j=1;i<=n && j<=m;i++)
        {
            int pos = lower_bound(b+1,b+tot+1,a[i])-b;
            if(!pre[pos])
            {
                add(i,b[pos]);
                pre[pos] = i;
            }
            else
            {
                add(pre[pos],-b[pos]);
                add(i,b[pos]);
                pre[pos] = i;
            }
            while(j<=m && q[j].r==i)
            {
                res[q[j].id] = ask(q[j].r)-ask(q[j].l-1);
                j++;
            }
        }
        for(int i=1;i<=m;i++)
            printf("%lld\n",res[i]);
    }

    return 0;
}