bzoj 1106 [POI2007]立方体大作战tet 树状数组优化

[POI2007]立方体大作战tet

Time Limit: 10 Sec  Memory Limit: 162 MB
Submit: 821  Solved: 601
[Submit][Status][Discuss]

Description

　　一个叫做立方体大作战的游戏风靡整个Byteotia。这个游戏的规则是相当复杂的，所以我们只介绍他的简单规
则：给定玩家一个有2n个元素的栈，元素一个叠一个地放置。这些元素拥有n个不同的编号，每个编号正好有两个
元素。玩家每次可以交换两个相邻的元素。如果在交换之后，两个相邻的元素编号相同，则将他们都从栈中移除，
所有在他们上面的元素都会掉落下来并且可以导致连锁反应。玩家的目标是用最少的步数将方块全部消除。

Input

　　第一行包含一个正整数n(1<=n<=50000)。接下来2n行每行一个数ai，从上到下描述整个栈，保证每个数出现且
仅只出现两次(1<=ai<=n)。初始时，没有两个相同元素相邻。并且保证所有数据都能在1000000步以内出解。

Output

　　第一行包含一个数m，表示最少的步数。

Sample Input

样例输入1
5
5
2
3
1
4
1
4
3
5
2
样例输入2
3
1
2
3
1
2
3

Sample Output

样例输出1
2
样例输出2
3

HINT

 

Source

 题解：其实就是暴力吧，如果直接暴力复杂度不对，所以用树状数组优化

 

 #include<iostream>

 #include<cstdio>

 #include<cstdlib>

 #include<algorithm>

 #include<cmath>

 #include<cstring>

 #define N 100007

 using namespace std;

 inline int read()

 {

     int x=,f=;char ch=getchar();

     while(ch<''||ch>''){if(ch=='-')f=-;ch=getchar();}

     while(ch>=''&&ch<=''){x=x*+ch-'';ch=getchar();}

     return x*f;

 }

 int n,ans;

 int a[N],t[N];

 int l[N];

 int query(int x)

 {

     int sum=;

     for(int i=x;i;i-=i&(-i))

         sum+=t[i];

     return sum;

 }

 void add(int x,int val)

 {

     for(int i=x;i<=*n;i+=i&(-i))

         t[i]+=val;

 }

 int main()

 {

     n=read();

     for(int i=;i<=*n;i++)

         a[i]=read();

     for(int i=;i<=*n;i++)

         if(!l[a[i]])add(i,),l[a[i]]=i;

         else

         {

             ans+=query(i)-query(l[a[i]]);

             add(l[a[i]],-);

         }

     printf("%d",ans);

 }