题目地址: 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;
}