hdu_3874 Necklace

原題鏈接：http://acm.hdu.edu.cn/showproblem.php?pid=3874

分析：線段樹，區間內不重複數的和。

           1、   用vis[]數組記錄將要插入的數是否在線段樹中，在的話就刪除（在相應的位置減去這個數）原來位置的值，將這個值插到現在的位置。

    如：1 1 1 2 3 5  插入第二個1的是好，第一個1在線段樹[1,1]的位置，這是先刪除[1,1]位置的1，再插入第二個1.

           2、 詢問之前先對所有的詢問數據按右端點升序排序，這樣就可以防止不斷刪除、插入的混亂（可以模擬一次，看看效果）。

我的代碼：

#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
#define MAXN 50005
#define MAXM 200005
typedef __int64 LL;
struct Node
{
    int l,r;
    LL sum;
}segtree[MAXN<<2];
struct piont
{
    int x,y;
    int i;
    bool operator <(const piont & a)const{
        return y<a.y;
    }
}cood[MAXM];
const int NUM=1000005; //最大的價值。
int p[MAXN];  //輸入的項鍊價值。
int vis[NUM];//表示M這個數在線段樹中得位置。
LL ans[MAXM]; //存放每次詢問的答案。

void build(int i,int s,int e)
{
    segtree[i].l=s;
    segtree[i].r=e;
    segtree[i].sum=0;
    if(s==e) return ;
    int mid=(s+e)>>1;
    build( 2*i, s, mid);
    build( 2*i+1,mid+1, e);
}
void Modfiy(int i,int add,int y) //在y的位置更新sum的值爲ADD，插入爲+，刪除爲-；
{
    if(y==segtree[i].l&&y==segtree[i].r)
    {
        segtree[i].sum+=add;
        return ;
    }
    int mid=(segtree[i].l+segtree[i].r)>>1;
    if(y<=mid)
    {
        Modfiy(2*i, add, y);
    }
    else
    {
        Modfiy(2*i+1, add, y);
    }
    segtree[i].sum+=add;
}
__int64 query(int i,int x,int y)
{
    if(segtree[i].l==x&&y==segtree[i].r) return segtree[i].sum;
    int mid=(segtree[i].l+segtree[i].r)>>1;
    if(y <= mid)
    {
      return   query( 2*i, x, y);
    }
    else if(x>mid) return query( 2*i+1, x, y);
    else
    {
        return  query( 2*i, x, mid)+query( 2*i+1, mid+1, y);
    }
}

int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        int n,m;
        scanf("%d",&n);
        for(int i=1;i<=n;i++) scanf("%d",p+i);
        scanf("%d",&m);
        for(int i=1;i<=m;i++)
        {
            scanf("%d%d",&cood[i].x,&cood[i].y);
            cood[i].i=i;
        }
        sort(cood+1,cood+1+m);
        build(1,1,n);
        memset(vis,0,sizeof(vis));
        int j=1; //輸入的下標。
        for(int i=1;i<=m;i++)
        {
            while(j<=cood[i].y)  //還沒有到詢問值。
            {//p[j]這個數已經在線段樹中，vis[i]=j,i這個在線段樹的j位置，。
                if(vis[p[j]])   Modfiy(1,-p[j],vis[p[j] ]);
                Modfiy(1,p[j],j);
                vis[p[j]]=j; //表示插到了j 這個點。
                j++;
            }
           ans[cood[i].i]= query(1,cood[i].x,cood[i].y);
        }
        for(int i=1;i<=m;i++)
        {
            printf("%I64d\n",ans[i]);
        }



    }


    return 0;
}

總結：這算是簡單的離散化，線段樹了。