使用段树解决问题

时间:2015-01-09 17:48:42

标签: java algorithm segment-tree

给定序列A的N(N <= 50000)整数在-10000和10000之间。在此序列中,您必须应用M(M <= 50000)运算: 修改序列中的第i个元素或者给定x y print max {Ai + Ai + 1 + .. + Aj | x&lt; = i&lt; = j&lt; = y}。

Problem Link



我正在使用Segment Tree但是我没有得到正确的输出,请帮助我在哪里犯了错误 的 CODE:

制作一棵树:

public static void maketree(int current , int a , int b ,int[] arr){

      if(b<a) return;

      if(b==a) {dp[current] = arr[a]; return ;}

      maketree(2*current, a, (a+b)/2, arr);

      maketree(2*current+1,1+ (a+b)/2, b, arr);

      if(dp[2*current]>0 && dp[2*current+1]>0) dp[current] = dp[2*current] + dp[2*current+1];
      else if(dp[2*current]>dp[2*current+1]) dp[current] = dp[2*current]; 
      else  dp[current] = dp[2*current+1]; 


}

更新功能

public static void update(int current , int a , int b , int c , int value){

      if(a>b || c<a || c>b) return ;

      if(a==b){ dp[current] = value; return ; }

      update(2*current, a, (a+b)/2, c, value);

      update(2*current+1, (b+a)/2 +1, b, c, value);

      if(dp[2*current]>0 && dp[2*current+1]>0) dp[current] = dp[2*current] + dp[2*current+1];
      else if(dp[2*current]>dp[2*current+1]) dp[current] = dp[2*current]; 
      else  dp[current] = dp[2*current+1]; 



}

查询功能:

public static int query(int current , int a , int b , int i , int j){
        int ans =0;


        if(a>j || b<i || a>b) return Integer.MIN_VALUE;

        if(a>=i && b<=j) return dp[current];

        int x = query(2*current, a, (a+b)/2, i, j);
        int y = query(2*current+1, (a+b)/2 +1, b, i, j);

       if(x>0 && y>0) ans= x+y;
       else if(x>y) ans = x;
       else ans =y;





        return ans; 



}

我不知道我在哪里弄错了请帮忙,dp array所需的存储容量是什么,即dp的大小

2 个答案:

答案 0 :(得分:1)

当你合并两个节点时,它可能就像下面给出的那样执行任何简单的例子,这样你就能感受到它:)

void merge(节点a,节点b)
{

    sum = a.sum + b.sum;
    pre = max(a.pre , (a.sum + b.pre));
    suf = max(b.suf , (b.sum + a.suf));
    result = max(a.suf + b.pre,max(a.result , b.result));

}

答案 1 :(得分:0)

这是非常复杂的imo ...

int tree[1 << 17]; // 2 ^ 17 >= N * 2
int M = 1; //base of tree or sth i dont remember english name

int query(int L, int R){
  int res = -10000; //minimum possible value in array
  L += M - 1;
  R += M - 1;
  while(L <= R){
    if(L % 2 == 1) res = max(res, tree[L++];
    if(R % 2 == 0) res = max(res, tree[R++];
    L /= 2;
    R /= 2;
  }
  return res;
}

void update(int v, int value){
  v += M - 1;
  tree[v] = value;
  while(v > 0){
    v /= 2;
    tree[v] = max(tree[v * 2], tree[v * 2 + 1]);
  }
}

void make_tree(){
  int n;
  cin >> n;
  while(M < n) M *= 2; // M is half of the size of tree
  for(int i = 0;i < n;i++)
    cin >> tree[i + M]; // just reading input to tree;
  for(int i = M - 1;i > 0;i--) // first update for all nodes other than leafs
    tree[i] = max(tree[i * 2], tree[i * 2 + 1]); 
}