树状数组练习：POJ 2481(JAVA)

 关于树状数组，请参考：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. }