1. 程式人生 > >[BZOJ4994] [Usaco2017 Feb]Why Did the Cow Cross the Road III(樹狀數組)

[BZOJ4994] [Usaco2017 Feb]Why Did the Cow Cross the Road III(樹狀數組)

樹狀數組 else cow why 貢獻 return cout inline cross

傳送門

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

  

[BZOJ4994] [Usaco2017 Feb]Why Did the Cow Cross the Road III(樹狀數組)