2016 UESTC Training for Data Structures R - Japan CDOJ 383 树状数组 逆序对

R - Japan

坐标有n个城市，右边有m个城市，城市都是按序号排序的，然后有K条连线，然后问连线的交点有多少个

也是类似于逆序数，不过跟E题不同的是不用考虑在左端点或者右端点的交点，只考虑交在中间的，也不用离散化

首先我们把每条线段按左端点升序排序，左端点相同按右端点升序排序

然后因为我们只数交点在中间的，所以我们首先找到同一个左端点的序号范围，假设当前为[i,j)，序号编号是从0开始，所以前面有i条直线，然后我们对于这个区间的每个，查询前面有多少个比这条直线右端点大的（i减去前面有多少小于等于的就行，树状数组维护），

然后查询之后，我们更新[i,j)这些直线，为什么这样呢，因为想排除右端点不同，查询时只需查询大于右端点的直线数即可，但是想要排除左端点不同，那就只能先查询，然后再更新，唔，具体可见代码

当然，还是发现数据比较弱，或者对于这块儿也没有明确去要求吧，反正之前一个不严格的程序也能AC

代码：

#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <algorithm>
using namespace std;
#define ll long long
#define maxn 10005
#define maxk 1000005
struct Line
{
	ll y1, y2;
	bool operator < (Line x)const
	{
		return y1 < x.y1 || (y1 == x.y1&&y2 < x.y2);
	}
}line[maxk];
int N, M, K;
ll tree[maxn];
void update(int i, ll val)
{
	if (i == 0)
		return;
	while (i <= M)
	{
		tree[i] += val;
		i += i&(-i);
	}
}
ll query(int i)
{
	ll sum = 0;
	while (i)
	{
		sum += tree[i];
		i -= i&(-i);
	}
	return sum;
}
int main()
{
	//freopen("input.txt", "r", stdin);
	int T;
	scanf("%d", &T);
	for (int kase = 1; kase <= T; ++kase)
	{
		memset(tree, 0, sizeof(ll)*maxn);
		for (int i = 0; i < K; ++i)
			line[i].y1 = 0, line[i].y2 = 0;
		scanf("%d%d%d", &N, &M, &K);
		for (int i = 0; i < K; ++i)
		{
			scanf("%lld%lld", &line[i].y1, &line[i].y2);
		}
		sort(line, line + K);
		ll ans = 0, cnt = 0;
		int last = 0;
		for (int i = 0; i < K;)
		{
			int j;
			for (j = i + 1; j < K; ++j)
			{
				if (line[j].y1 != line[i].y1)
					break;
			}
			for (int k = i; k < j; ++k)
			{
				ans += i - query(line[k].y2);
				//printf("%d %lld    q %lld\n", k, ans, query(line[k].y2));
			}
			for (int k = i; k < j; ++k)
			{
				update(line[k].y2, 1);
			}
			i = j;
			//update(line[i].y2, 1);
			//ans += i + 1 - query(line[i].y2);
		}
		printf("Test case %d: %lld\n", kase, ans);
	}
	//system("pause");
	//while (1);
	return 0;
}