【USACO 2017FEB】 Why Did the Cow Cross the Road III

【题目链接】

点击打开链接

【算法】

树状数组

【代码】

#include<bits/stdc++.h>
using namespace std;
#define MAXN 100010
 
int n,i,x;
long long ans;
int pos[MAXN];
long long c[MAXN];
 
inline int lowbit(int x)
{
    return x & (-x);
}
inline void modify(int pos,long long val)
{
    int i;
    for (i = pos; i <=  * n; i += lowbit(i)) c[i] += val;
}
inline long long query(int pos)
{
    int i;
        long long ans = ;
    for (i = pos; i; i -= lowbit(i)) ans += c[i];
    return ans;
}
inline long long query(int l,int r)
{
        return query(r) - query(l-);
}
 
int main() {
 
        scanf("%d",&n);
        for (i = ; i <=  * n; i++)
        {
                scanf("%d",&x);
                if (!pos[x])
                {
                        pos[x] = i;
                        modify(i,);
                }    else
                {
                        ans += query(pos[x]+,i-);
                        modify(pos[x],-);
                }
        }
        printf("%lld\n",ans);
 
        return ;
 
}