poj3067 树状数组解决二维逆序对

最新推荐文章于 2020-01-11 19:39:41 发布

D.Y.T

最新推荐文章于 2020-01-11 19:39:41 发布

阅读量394

点赞数 1

分类专栏： Coding 文章标签：树状数组

本文链接：https://blog.csdn.net/qq_43279710/article/details/98938256

版权

Coding 专栏收录该内容

123 篇文章 0 订阅

订阅专栏

一直在学习二维逆序对的解决方法，然后并没有搞明白，这是第一道自己搞出来的模板题。

POJ3067 JAPAN
题解：
设东部的城市为 $e_i$ ，西部的城市为 $w_i$ ，那么如果连接 $e_1$ $w_2$ 和连接 $e_2$ $w_1$ 的两条道路想形成交叉，必然有当 $e_1 < e_2$ 时， $w_1 > w_2$ ，也就是说，题意所求的交叉数量一定是在 $e$ 有序的情况下，求 $w$ 的逆序对数，所以我们先对 $e$ 排序，在用树状数组对 $w$ 求逆序对数即可。

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
using namespace std;
typedef long long ll;

const int N = 10100;

int n, m, k;
ll num[N], c[N];

struct road{
	ll e, w;
}r[1010100];

bool cmp(road a, road b){
	if(a.e != b.e)
		return a.e < b.e;
	return a.w <= b.w;	
}
void add(int x, int y){
	for(;x <= m; x += x & -x) c[x]+=y;
}

ll ask(int x){
	ll ans = 0;
	for(;x;x -= x &-x) ans+=c[x];
	return ans;
}

int main(){
	int t;
	scanf("%d", &t);
	int cas = 0;
	while(t--){
		scanf("%d%d%d",&n,&m,&k);
		memset(num, 0, sizeof(num));
		memset(c, 0, sizeof(c));
		for(int i = 1;i <= k; i++){
			scanf("%lld%lld", &r[i].e, &r[i].w);
		}
		sort(r+1, r+k+1, cmp);
		ll ans = 0;
		for(int i = k; i; i--){
			ans += ask(r[i].w - 1);
			add(r[i].w, 1);
		}
		printf("Test case %d: %lld\n", ++cas, ans);
	}
	return 0;
}