[BZOJ4994] [Usaco2017 Feb]Why Did the Cow Cross the Road III（树状数组）

传送门

 

1.每个数的左右位置预处理出来，按照左端点排序，因为左端点是从小到大的，我们只需要知道每条线段包含了多少个前面线段的右端点即可，可以用树状数组

2.如果 ai < bj < bi， 也就是说在两个i之间只有一个j那么对答案的贡献为1，所以可以用树状数组，当第一次出现一个数的时候，那个位置+1，当第二次出现的时候，第一次出现的位置再-1，也就是把它对答案的贡献去掉，同样是树状数组统计答案

 

#include <cstdio>
#include <iostream>
#define N 200001

int n, ans;
int sum[N], vis[N];

inline int read()
{
	int x = 0, f = 1;
	char ch = getchar();
	for(; !isdigit(ch); ch = getchar()) if(ch == '-') f = -1;
	for(; isdigit(ch); ch = getchar()) x = (x << 1) + (x << 3) + ch - '0';
	return x * f;
}

inline int query(int x)
{
	int ret = 0;
	for(; x; x -= x & -x) ret += sum[x];
	return ret;
}

inline void add(int x, int d)
{
	for(; x <= n; x += x & -x) sum[x] += d;
}

int main()
{
	int i, x;
	n = read() * 2;
	for(i = 1; i <= n; i++)
	{
		x = read();
		if(!vis[x])
			vis[x] = i, add(i, 1);
		else
			ans += query(i) - query(vis[x]), add(vis[x], -1);
	}
	std::cout << ans;
	return 0;
}