题意: 日本的东西海岸分别有 M, N 座城市,从北至南依次编号,现在东西海岸的某些城市之间修建公路,求所有交点的数量。
题解: 若 a, b 两公路存在交点的话,必有 a.east < b.east, a. west > b.west 或者 a.east > b.east, a.west < b.west。 对每一条公路的 east 降序排列,east 相同则 west 也按降序排列。这样一来,只需求 west 的逆序数。用树状数组。
#include <algorithm>
#include <iostream>
using namespace std;
#define N 1000005
int n, m;
__int64 c[N];
struct item
{
int w, e;
} node[N];
bool cmp ( item x, item y )
{
if ( x.e != y.e )
return x.e > y.e;
return x.w > y.w;
}
int lowbit ( int x )
{
return x & ( -x );
}
void update ( int x )
{
int t = (m > n ? m : n);
for ( int i = x; i <= t; i += lowbit(i) )
c[i]++;
}
int getSum ( int x )
{
int sum = 0;
for ( int i = x; i > 0; i -= lowbit(i) )
sum += c[i];
return sum;
}
int main()
{
int t, k, temp, i, test = 0;
__int64 ans;
scanf("%d",&t);
while ( t-- )
{
ans = 0;
memset(c,0,sizeof(c));
scanf("%d%d%d",&n,&m,&k);
for ( i = 1; i <= k; ++i )
scanf("%d%d", &node[i].e, &node[i].w);
sort(node+1,node+1+k,cmp);
for ( i = 1; i <= k; ++i )
{
if ( node[i-1].e == node[i].e && node[i-1].w == node[i].w )
ans += temp; /* temp 代表前一条公路以其他公路的交点数 */
else
{
temp = getSum(node[i].w-1);
ans += temp;
}
update(node[i].w);
}
printf("Test case %d: %I64d\n", ++test, ans);
}
return 0;
}