树状数组练习：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过的代码:

import java.io.StreamTokenizer;
import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.io.PrintWriter;
import java.io.OutputStreamWriter;
import java.io.IOException;
import java.util.Arrays;

class TNode implements Comparable{//表示一头牛
int s;//牛吃草的左端点
int e;//右端点
int label;//牛的序号
public int compareTo(Object o) {
int v=((TNode)o).e;
if(this.e!=v) //按右端点降序排列
return v-this.e;
return this.s-((TNode)o).s;//右端点相等,按左端点升序排序
}

public String toString(){
return ("["+s+","+e+"]");
}
}

public class Main{//
static int N=100015;
TNode[] cow;
int cal[];//树状数组
int res[];//res[i]表示比牛i强壮的牛的个数
int maxn;//右端点的最大值

public Main(){

}

private int lowbit(int t){//计算cal[t]展开的项数
return t&(-t);
}

private int Sum(int k){ //求前k项的和.
int sum=0;
while(k>0){
sum+=cal[k];
k=k-lowbit(k);
}
return sum;
}

private void update(int i,int x){ //增加某个元素的大小,给某个节点 i 加上 x
while(i<=maxn){
cal[i]=cal[i]+x; //更新父节点
i=i+lowbit(i);
}
}




public static void main(String args[]) throws IOException{
Main ma=new Main();
ma.go();
}

public void go() throws IOException{

StreamTokenizer st = new StreamTokenizer(new BufferedReader(
new InputStreamReader(System.in)));
PrintWriter out = new PrintWriter(new OutputStreamWriter(System.out));

while(true) {
st.nextToken();
int n= (int) st.nval; //牛的个数
if(n==0) break;

cow=new TNode[N];
cal=new int[N];
res=new int[N];
for(int i=0;i
cow[i]=new TNode();
// cal[i]=0;
}
for(int i=0;i//读入牛的吃草区间
st.nextToken();
cow[i].s=(int) st.nval;
st.nextToken();
cow[i].e=(int) st.nval;
cow[i].s++;
cow[i].e++;
cow[i].label=i;//牛的原始序号
if(cow[i].e>maxn) maxn=cow[i].e;//最大右端点
}

Arrays.sort(cow);//排序

// for(int i=0;i
// System.out.println(cow[i]);

for(int i=0;i
if(i!=0&&cow[i].s==cow[i-1].s&&cow[i].e==cow[i-1].e)//两头牛有相同的吃草区间
res[cow[i].label]=res[cow[i-1].label];//它们有相同的答案
else res[cow[i].label]=Sum(cow[i].s);//统计比cow[i].label这头牛强的牛的数目
update(cow[i].s,1);//更新
}
System.out.printf("%d",res[0]);
for(int i=1;i" %d",res[i]);
System.out.println();
}
}
}