import java.util.Scanner; /*線段樹空間計算程序 Power By:Comzyh*/ class segment {//線段樹節點 int b,e; } public class SegmentTree{//線段樹,用數組實現 static int M=5000000; segment seg[]; int Nnode;//節點數 int LastNode;//最後一個節點 public SegmentTree(){ seg=new segment[M]; for(int i=0;i<M;i++) seg[i]=new segment(); } void build(int b, int e, int root){//構造線段樹 Nnode++; if (root>LastNode) LastNode=root; seg[root].b=b; seg[root].e=e; if (b==e) return; int mid =(b+e)>>1; build(b,mid,root<<1); build(mid+1,e,(root<<1)+1); } public int getNnode(){ return Nnode; } public int getLastNode(){ return LastNode; } public static void main(String args[]){ Scanner in=new Scanner(System.in); int N; while (true){ System.out.printf("請輸入區間長度:\n"); N=in.nextInt(); if (N==0) break; SegmentTree st=new SegmentTree(); st.build(1,N,1); System.out.printf("線段樹構造完成, 共有%d 節點,最後一個節點是: %d\n",st.getNnode(),st.getLastNode()); //注意:節點個數老是2N-1 } } }
運行:
C:\ex>java SegmentTree
請輸入區間長度:
5
線段樹構造完成, 共有9 節點,最後一個節點是: 9
請輸入區間長度:
10
線段樹構造完成, 共有19 節點,最後一個節點是: 25
請輸入區間長度:
100000
線段樹構造完成, 共有199999 節點,最後一個節點是: 262141 node