【BZOJ1106】【POI2007】立方体大作战tet(树状数组+贪心)

贪心策略:每加入一个数,如果之前已经存在它了,就直接交换

因此我们需要维护距离 就用树状数组好了

注意是2n

#include<bits/stdc++.h>
#define N 100005
using namespace std;
int n,tree[N],pre[N],ans;
inline int lowbit(int x)
{
    return x&(-x);
}
inline void update(int x,int del)
{
    for(int i=x;i<=2*n;i+=lowbit(i))    tree[i]+=del;
} 
inline int query(int x)
{
    int ans=0;
    for(int i=x;i;i-=lowbit(i)) ans+=tree[i];
    return ans;
}
int main()
{
    scanf("%d",&n);
    for(int i=1,x;i<=2*n;i++)
    {
        scanf("%d",&x);
        if(!pre[x])
        {
            pre[x]=i;
            update(i,1);
        }
        else
        {
            ans+=query(i)-query(pre[x]);
            update(pre[x],-1);
        }
    }
    cout<<ans;
    return 0;
}

猜你喜欢

转载自www.cnblogs.com/Patrickpwq/p/9853312.html