关于树状数组,请参考:http://128kj.iteye.com/blog/1743633

POJ 2481题意:
有n头牛(编号为1~n),每一头牛都有一个吃草区间[S, E],如果对于牛i和牛j来说,它们的吃草区间满足下面的条件则证明牛i比牛j强壮:Si <= Sj and Ej <= Ei and Ei - Si > Ej - Sj。现在已知每一头牛的吃草区间,要求输出每头牛有几头牛比其强壮。
其中:1 <= N <= 100000, 0 <= S < E <= 100000

样例:

Sample Input

3 (3头牛)
1 2 牛的[s,e]值
0 3
3 4
0

Sample Output

1 0 0 (比第一头牛强的有一头,比第二头牛强的没有,比第三头牛强的没有)

分析:
建一个全0的树状数组,然后将牛一个一个插入树状数组,当然必须先对e按照降序排列,每加入一只牛,当前已经加入树状数组的牛的s如果比这只牛小,那么那些牛就更强壮,所以这是在树状数组里的求和问题,只要求出在插入s之前已插入了多少头牛(求和)。另外,由于排序后,顺序和开始给出数据时的顺序不同,所以需要记录一下开始给出数据时的顺序。

下面是AC过的代码:

Java代码 复制代码 收藏代码
  1. import java.io.StreamTokenizer;
  2. import java.io.BufferedReader;
  3. import java.io.InputStreamReader;
  4. import java.io.PrintWriter;
  5. import java.io.OutputStreamWriter;
  6. import java.io.IOException;
  7. import java.util.Arrays;
  8.  
  9. class TNode implements Comparable{//表示一头牛
  10. int s;//牛吃草的左端点
  11. int e;//右端点
  12. int label;//牛的序号
  13. public int compareTo(Object o) {
  14. int v=((TNode)o).e;
  15. if(this.e!=v) //按右端点降序排列
  16. return v-this.e;
  17. return this.s-((TNode)o).s;//右端点相等,按左端点升序排序
  18. }
  19.  
  20. public String toString(){
  21. return ("["+s+","+e+"]");
  22. }
  23. }
  24.  
  25. public class Main{//
  26. static int N=100015;
  27. TNode[] cow;
  28. int cal[];//树状数组
  29. int res[];//res[i]表示比牛i强壮的牛的个数
  30. int maxn;//右端点的最大值
  31.  
  32. public Main(){
  33.  
  34. }
  35.  
  36. private int lowbit(int t){//计算cal[t]展开的项数
  37. return t&(-t);
  38. }
  39.  
  40. private int Sum(int k){ //求前k项的和.
  41. int sum=0;
  42. while(k>0){
  43. sum+=cal[k];
  44. k=k-lowbit(k);
  45. }
  46. return sum;
  47. }
  48.  
  49. private void update(int i,int x){ //增加某个元素的大小,给某个节点 i 加上 x
  50. while(i<=maxn){
  51. cal[i]=cal[i]+x; //更新父节点
  52. i=i+lowbit(i);
  53. }
  54. }
  55.  
  56.  
  57.  
  58.  
  59. public static void main(String args[]) throws IOException{
  60. Main ma=new Main();
  61. ma.go();
  62. }
  63.  
  64. public void go() throws IOException{
  65.  
  66. StreamTokenizer st = new StreamTokenizer(new BufferedReader(
  67. new InputStreamReader(System.in)));
  68. PrintWriter out = new PrintWriter(new OutputStreamWriter(System.out));
  69.  
  70. while(true) {
  71. st.nextToken();
  72. int n= (int) st.nval; //牛的个数
  73. if(n==0) break;
  74.  
  75. cow=new TNode[N];
  76. cal=new int[N];
  77. res=new int[N];
  78. for(int i=0;i<N;i++){
  79. cow[i]=new TNode();
  80. // cal[i]=0;
  81. }
  82. for(int i=0;i<n;i++) {//读入牛的吃草区间
  83. st.nextToken();
  84. cow[i].s=(int) st.nval;
  85. st.nextToken();
  86. cow[i].e=(int) st.nval;
  87. cow[i].s++;
  88. cow[i].e++;
  89. cow[i].label=i;//牛的原始序号
  90. if(cow[i].e>maxn) maxn=cow[i].e;//最大右端点
  91. }
  92.  
  93. Arrays.sort(cow);//排序
  94.  
  95. // for(int i=0;i<n;i++)
  96. // System.out.println(cow[i]);
  97.  
  98. for(int i=0;i<n;i++) {
  99. if(i!=0&&cow[i].s==cow[i-1].s&&cow[i].e==cow[i-1].e)//两头牛有相同的吃草区间
  100. res[cow[i].label]=res[cow[i-1].label];//它们有相同的答案
  101. else res[cow[i].label]=Sum(cow[i].s);//统计比cow[i].label这头牛强的牛的数目
  102. update(cow[i].s,1);//更新
  103. }
  104. System.out.printf("%d",res[0]);
  105. for(int i=1;i<n;i++) System.out.printf(" %d",res[i]);
  106. System.out.println();
  107. }
  108. }
  109. }