PKU3067 Japan - 树状数组

题目描述:

城市被分为东西两个部分,两岸城市之间有高速公路连接。给出高速公路连接情况,判断有公路一共有多少个交点。

分析:

类似二部图匹配的模型。

设公路左边城市为x,右边城市为y。首先按照x城市的编号从大到小将公路a[]排序。

对于公路a[i]交点个数为,0到i-1之间的公路y<a[i].y的个数。(描述不清¥%#&……不说了,代码很明白)

这里是典型的树状数组的应用。

要注意两个地方:第一,公路可能有很多条,数组要开到10^6;第二,求和可能超int,要用int64。

 

  1. /*
  2. PKU3067 Japan
  3. */
  4. #include <stdio.h>
  5. #include <stdlib.h>
  6. #include <memory.h>
  7. #define clr(a) memset(a,0,sizeof(a))
  8. #define int64 __int64
  9. #define I64d "%I64d"
  10. #define M 1000000
  11. #define N 3000
  12. typedef struct{
  13.     int x,y;
  14. }Node;
  15. int n,m;
  16. Node a[M];
  17. int c[N];
  18. int cmp(const void *A,const void *B){
  19.     Node a = *(Node*)A;
  20.     Node b = *(Node*)B;
  21.     if(b.x==a.x) return b.y-a.y;
  22.     return b.x-a.x;
  23. }
  24. int lowbit(int x){
  25.     return x&(-x);
  26. }
  27. void Add(int i){
  28.     while(i<=n){
  29.         c[i]++;
  30.         i+=lowbit(i);
  31.     }
  32. }
  33. int64 Sum(int i){
  34.     int64 s=0;
  35.     while(i>0){
  36.         s+=c[i];
  37.         i-=lowbit(i);
  38.     }
  39.     return s;
  40. }
  41. int main()
  42. {
  43.     int T,Tn;
  44.     
  45.     scanf("%d",&Tn);
  46.     
  47.     for(T=1;T<=Tn;T++){
  48.         int i,j,k;
  49.         //input
  50.         scanf("%*d%d%d",&n,&m);
  51.         for(i=0;i<m;i++){
  52.             scanf("%d%d",&a[i].x,&a[i].y);
  53.         }
  54.         //sort
  55.         qsort(a,m,sizeof(a[0]),cmp);
  56.         
  57.         //work
  58.         int64 sum=0;
  59.         clr(c);
  60.         
  61.         for(k=0;k<m;k++){
  62.             i=a[k].y;
  63.             Add(i);
  64.             sum+=Sum(i-1);
  65.         }
  66.         //output
  67.         printf("Test case %d: "I64d"/n",T,sum);
  68.     }
  69.     
  70.     return 0;
  71. }

你可能感兴趣的:(PKU3067 Japan - 树状数组)