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);
 }