POJ 3067 - Japan

 

题目地址： http://poj.org/problem?id=3067

 

典型的树状数组题目。

 

    跟小学时做的连线题很像，问共有多少个交点。两点间只能用线段连，并且假设不会有3条及以上的线段交于同一点。

    假设两个点 （x1，y1）、（x2，y2）

    假若 y2 > y1 ，那么此时，只要存在 x1 < x2 ，那么就一定存在交点。

    即其中一点在另一点的左上角。

    所以我们先对 y从大到小排序，若y相等则对x从大到小排序。

    于是，对排序完成后的点集。从前往后遍历。

每次遍历：update(xi，1) ，横坐标为xi的数，数量加1 。 然后 query(xi - 1) ，其结果代表当前已记录的、小于xi的横坐标总数，即得到该点左上角共有多少点。

 

    简单来说，就是用树状数组来hash得到某个范围内的数的总数。而这个范围内某些数的数量是不断更新的。

 

#include<iostream>
#include<cstdio>
#include<algorithm>

using namespace std;

int tree[10100];

struct Point{
	int x,y;
}p[1001000];

int cmp(Point a,Point b){
	if(a.y==b.y) return a.x>b.x;
	return a.y>b.y;
}

int query(int x){
	int sum=0;
	while(x>0){
		sum+=tree[x];
		x-=x&-x;
	}
	return sum;
}
void update(int x,int n){
	while(x<=n){
		tree[x]++;
		x+=x&-x;
	}
}

int main(){
	int t,tt;
	int n,m,k;
	int sum,i;
	scanf("%d",&t);
	for(tt=1;tt<=t;tt++){
		scanf("%d%d%d",&n,&m,&k);
		for(i=1;i<=n;i++)
			tree[i]=0;
		sum=0;
		for(i=0;i<k;i++)
			scanf("%d%d",&p[i].x,&p[i].y);
		sort(p,p+k,cmp);
		for(i=0;i<k;i++){
			sum+=query(p[i].x-1);
			update(p[i].x,n);
		}
		printf("Test case %d: %d\n",tt,sum);
	}
	return 0;
}