bzoj 4994: [Usaco2017 Feb]Why Did the Cow Cross the Road III

最新推荐文章于 2020-03-29 11:35:37 发布

fyc_kabuto

最新推荐文章于 2020-03-29 11:35:37 发布

阅读量501

点赞数 1

分类专栏：树状数组

本文链接：https://blog.csdn.net/qq_36808030/article/details/77746074

版权

树状数组专栏收录该内容

11 篇文章 0 订阅

订阅专栏

题意：

给定长度为2N的序列，1~N各处现过2次，i第一次出现位置记为ai，第二次记为bi，求满足ai

题解：

大水题，记录下每个数的第一次出现位置，第一次访问时在其位置加一，第二次减回一，且求区间和，累加就是答案了。
树状数组维护下就好啦。
code：

#include<cstdio>
#include<cstdlib>
#include<iostream>
#include<cstring>
#define LL long long
using namespace std;
int p[100010],tr[100010],n;
LL ans=0;
int lowbit(int x){return x&(-x);}
void change(int k,int c)
{
    for(int i=k;i<=2*n;i+=lowbit(i))
        tr[i]+=c;
}
int get(int k)
{
    int ans=0;
    for(int i=k;i>=1;i-=lowbit(i))
        ans+=tr[i];
    return ans;
}
int main()
{
    scanf("%d",&n);
    memset(p,-1,sizeof(p));
    for(int i=1;i<=2*n;i++)
    {
        int x;scanf("%d",&x);
        if(p[x]==-1) p[x]=i,change(i,1);
        else
        {
            change(p[x],-1);
            ans=ans+(LL)(get(i)-get(p[x]));
        }
    }
    printf("%lld",ans);
}