算法训练 格子操作(线段树)-------------------------C语言——菜鸟级

/*问题描述
有n个格子,从左到右放成一排,编号为1-n。

共有m次操作,有3种操作类型:

1.修改一个格子的权值,

2.求连续一段格子权值和,

3.求连续一段格子的最大值。

对于每个2、3操作输出你所求出的结果。

输入格式
第一行2个整数n,m。

接下来一行n个整数表示n个格子的初始权值。

接下来m行,每行3个整数p,x,y,p表示操作类型,p=1时表示修改格子x的权值为y,p=2时表示求区间[x,y]内格子权值和,p=3时表示求区间[x,y]内格子最大的权值。

输出格式
有若干行,行数等于p=2或3的操作总数。

每行1个整数,对应了每个p=2或3操作的结果。

样例输入
4 3
1 2 3 4
2 1 3
1 4 3
3 1 4
样例输出
6
3
数据规模与约定
对于20%的数据n <= 100,m <= 200。

对于50%的数据n <= 5000,m <= 5000。

对于100%的数据1 <= n <= 100000,m <= 100000,0 <= 格子权值 <= 10000。*/

#include<stdio.h> 
#define N 40000001
#define MAX(a,b) a>b?a:b
typedef struct node
{long int max,sum;
}L;
L node[N];

 void f(long int now,long int l,long int r, long int le,long int ri,long int s)//修改 或初始附值 
 {        if(r==ri&&l==le)
          { 
             node[now].sum=s;
             node[now].max=s;
          }
          else
          {
            long int mid=(r+l)>>1;
            long int next=now<<1;
            if(ri<=mid)f(next,l,mid,le,ri,s);
            else f(next+1,mid+1,r,le,ri,s);
            node[now].sum=node[next].sum+node[next+1].sum;
            node[now].max=MAX(node[next].max,node[next+1].max);
          }
 }

long int outsum(long int now,long int l,long int r ,long int le,long int ri)//求区间和 
{
     if(r==ri&&l==le)return node[now].sum ;
       else
       {  long int mid=(r+l)>>1;
          long int next=now<<1;
          if(ri<=mid) return outsum(next,l,mid,le,ri);
           else if(mid<le) return outsum(next+1,mid+1,r,le,ri);
             else return outsum(next,l,mid,le,mid)+outsum(next+1,mid+1,r,mid+1,ri);
       }
}
 long int outmax(long int now,long int l,long int r,long int le,long int ri)//求区间最大值 
 { long int M;
        if(r==ri&&l==le)M=node[now].max;
        else
        { long int mid=(r+l)>>1;
          long  int next=now<<1;
          if(ri<=mid)M=outmax(next,l,mid,le,ri);
          else if(le>mid)M=outmax(next+1,mid+1,r,le,ri);
          else M=MAX(outmax(next,l,mid,le,mid),outmax(next+1,mid+1,r,mid+1,ri));
        }

    return M;

 }
int main()
{
 long int n,i,s,m,p,x,y;
 scanf("%ld%ld",&n,&m);

 for(i=1;i<=n;i++)
 { scanf("%ld",&s);
   f(1,1,n,i,i,s);
 }
for(i=0;i<m;i++)
{  scanf("%ld%ld%ld",&p,&x,&y);
   if(p==1)f(1,1,n,x,x,y);
   if(p==2)printf("%ld\n",outsum(1,1,n,x,y));
   if(p==3)printf("%ld\n",outmax(1,1,n,x,y));
}
return 0;
} 
阅读更多

扫码向博主提问

Five-菜鸟级

非学,无以致疑;非问,无以广识
  • 擅长领域:
  • C
  • ACM
  • lanqiaobei
去开通我的Chat快问
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页