洛谷P3372/poj3468（线段树lazy_tag）（询问区间和，支持区间修改）

洛谷P3372

 //线段树  询问区间和，支持区间修改
 #include <cstdio>

 using namespace std;

  struct treetype
  {
      int l,r;
      long long sum,d;
  };

 const int maxn=;

 treetype a[maxn<<];
 long long num[maxn];

 void build(int k,int l,int r)
 {
     a[k].l=l;a[k].r=r;a[k].d=;
     if (a[k].l==a[k].r)
     {
         a[k].sum=num[l];
         return;
     }
     int mid=(l+r)>>,i=k<<;
     build(i,l,mid);
     build(i+,mid+,r);
     a[k].sum=a[i].sum+a[i+].sum;
 }
 void pushdown(int k)
 {
     if (a[k].l==a[k].r) a[k].d=;
     if (a[k].d==) return;
     //int i=k<<1;long long t=a[i].r-a[i].l+1;a[i].sum+=t*a[k].d;a[i].d+=a[k].d;
     //i++;t=a[i].r-a[i].l+1;a[i].sum+=t*a[k].d;a[i].d+=a[k].d;
     int i=k<<;a[i].sum+=(a[i].r-a[i].l+)*a[k].d;a[i].d+=a[k].d;
     i++;a[i].sum+=(a[i].r-a[i].l+)*a[k].d;a[i].d+=a[k].d;
     a[k].d=;
 }
 void change(int k,int l,int r,long long d)
 {
     pushdown(k);
     if (l<=a[k].l && a[k].r<=r)
     {
         //long long t=a[k].r-a[k].l+1;
         a[k].sum+=(a[k].r-a[k].l+)*d;a[k].d=d;
         return;
     }
     int mid=(a[k].l+a[k].r)>>,i=k<<;
     if (l<=mid) change(i,l,r,d);
     if (mid<r) change(i+,l,r,d);
     a[k].sum=a[i].sum+a[i+].sum;
 }
 long long query(int k,int l,int r)
 {
     pushdown(k);
     if (l<=a[k].l && a[k].r<=r) return a[k].sum;
     int mid=(a[k].l+a[k].r)>>,i=k<<;
     long long ans=;
     if (l<=mid) ans=query(i,l,r);
     if (mid<r) ans+=query(i+,l,r);
     return ans;
 }

 int main()
 {
     int n,m;
     scanf("%d%d",&n,&m);
     for (int i=;i<=n;i++) scanf("%lld",&num[i]);
     build(,,n);
     for (int i=;i<=m;i++)
     {
         int p,x,y;
         scanf("%d%d%d",&p,&x,&y);
         if (p==)
         {
             long long k;
             scanf("%lld",&k);
             change(,x,y,k);
         }
         else printf("%lld\n",query(,x,y));
     }
     return ;
 }

poj3468

 //线段树  询问区间和，支持区间修改
 #include <cstdio>

 using namespace std;

  struct treetype
  {
      int l,r;
      long long sum,d;
  };

 const int maxn=;

 treetype a[maxn<<];
 long long num[maxn];

 void build(int k,int l,int r)
 {
     a[k].l=l;a[k].r=r;a[k].d=;
     if (a[k].l==a[k].r)
     {
         a[k].sum=num[l];
         return;
     }
     int mid=(l+r)>>,i=k<<;
     build(i,l,mid);
     build(i+,mid+,r);
     a[k].sum=a[i].sum+a[i+].sum;
 }
 void pushdown(int k)
 {
     if (a[k].l==a[k].r) a[k].d=;
     if (a[k].d==) return;
     int i=k<<;a[i].sum+=(a[i].r-a[i].l+)*a[k].d;a[i].d+=a[k].d;
     i++;a[i].sum+=(a[i].r-a[i].l+)*a[k].d;a[i].d+=a[k].d;
     a[k].d=;
 }
 void change(int k,int l,int r,long long d)
 {
     pushdown(k);
     if (l<=a[k].l && a[k].r<=r)
     {
         a[k].sum+=(a[k].r-a[k].l+)*d;a[k].d=d;
         return;
     }
     int mid=(a[k].l+a[k].r)>>,i=k<<;
     if (l<=mid) change(i,l,r,d);
     if (mid<r) change(i+,l,r,d);
     a[k].sum=a[i].sum+a[i+].sum;
 }
 long long query(int k,int l,int r)
 {
     pushdown(k);
     if (l<=a[k].l && a[k].r<=r) return a[k].sum;
     int mid=(a[k].l+a[k].r)>>,i=k<<;
     long long ans=;
     if (l<=mid) ans=query(i,l,r);
     if (mid<r) ans+=query(i+,l,r);
     return ans;
 }

 int main()
 {
     int n,m;
     scanf("%d%d",&n,&m);
     for (int i=;i<=n;i++) scanf("%lld",&num[i]);
     build(,,n);
     for (int i=;i<=m;i++)
     {
         char p;scanf("%c",&p);
         int x,y;
         scanf("%c%d%d",&p,&x,&y);
         if (p=='C')
         {
             long long k;
             scanf("%lld",&k);
             change(,x,y,k);
         }
         else printf("%lld\n",query(,x,y));
     }
     return ;
 }