zoukankan      html  css  js  c++  java
  • 线段树2

    题目描述

    如题,已知一个数列,你需要进行下面三种操作:

    1.将某区间每一个数乘上x

    2.将某区间每一个数加上x

    3.求出某区间每一个数的和

    输入输出格式

    输入格式:

    第一行包含三个整数N、M、P,分别表示该数列数字的个数、操作的总个数和模数。

    第二行包含N个用空格分隔的整数,其中第i个数字表示数列第i项的初始值。

    接下来M行每行包含3或4个整数,表示一个操作,具体如下:

    操作1: 格式:1 x y k 含义:将区间[x,y]内每个数乘上k

    操作2: 格式:2 x y k 含义:将区间[x,y]内每个数加上k

    操作3: 格式:3 x y 含义:输出区间[x,y]内每个数的和对P取模所得的结果

    输出格式:

    输出包含若干行整数,即为所有操作3的结果。

    输入输出样例

    输入样例#1: 复制
    5 5 38
    1 5 4 2 3
    2 1 4 1
    3 2 5
    1 2 4 2
    2 3 5 5
    3 1 4
    输出样例#1: 复制
    17
    2

    说明

    时空限制:1000ms,128M

    数据规模:

    对于30%的数据:N<=8,M<=10

    对于70%的数据:N<=1000,M<=10000

    对于100%的数据:N<=100000,M<=100000

    (数据已经过加强^_^)

    样例说明:

    故输出应为17、2(40 mod 38=2)


    就是基本的线段树操作,但是有几个地方要多注意下,也就是我犯错的地方。

    1 乘法优先级高于加法

    2 初始tag数组时不能用memset把mul全变成1,要在build中赋初始值

    3 long long记得最后要改好

    #include<iostream>
    #include<cstdio>
    #include<cstdlib>
    #include<cstring>
    #include<algorithm>
    
    #define ll long long
    #define il inline
    #define db double
    
    using namespace std;
    
    il int gi()
    {
      int x=0,y=1;
      char ch=getchar();
      while(ch<'0'||ch>'9')
        {
          if(ch=='-')
    	y=-1;
          ch=getchar();
        }
      while(ch>='0'&&ch<='9')
        {
          x=x*10+ch-'0';
          ch=getchar();
        }
      return x*y;
    }
    
    il ll gl()
    {
      ll x=0,y=1;
      char ch=getchar();
      while(ch<'0'||ch>'9')
        {
          if(ch=='-')
    	y=-1;
          ch=getchar();
        }
      while(ch>='0'&&ch<='9')
        {
          x=x*10+ch-'0';
          ch=getchar();
        }
      return x*y;
    }
    
    int n,mm;
    ll p;
    ll a[100045],sum[400045];
    ll add[400045],mul[400045];
    
    il void build(int rt,int l,int r)
    {
      mul[rt]=1;
      add[rt]=0;
      int m=(l+r)>>1;
      if(l==r)
        {
          sum[rt]=a[l];
          return;
        }
      build(rt<<1,l,m);
      build(rt<<1|1,m+1,r);
      sum[rt]=(sum[rt<<1]+sum[rt<<1|1])%p;
    } 
    
    il void pushdown(int rt,int l,int r)
    {
      int m=(l+r)>>1;
      sum[rt<<1]=(sum[rt<<1]*mul[rt]%p+add[rt]*(m-l+1)%p)%p;
      sum[rt<<1|1]=(sum[rt<<1|1]*mul[rt]%p+add[rt]*(r-m)%p)%p;
      mul[rt<<1]=mul[rt<<1]*mul[rt]%p;
      mul[rt<<1|1]=mul[rt<<1|1]*mul[rt]%p;
      add[rt<<1]=(add[rt<<1]*mul[rt]%p+add[rt])%p;
      add[rt<<1|1]=(add[rt<<1|1]*mul[rt]%p+add[rt])%p;
      mul[rt]=1;
      add[rt]=0;
    }
    
    il void update1(int rt,int l,int r,int L,int R,ll k)
    {
      if(L<=l&&R>=r)
        {
          sum[rt]=sum[rt]*k%p;
          mul[rt]=mul[rt]*k%p;
          add[rt]=add[rt]*k%p;
          return;
        }
      pushdown(rt,l,r);
      int m=(l+r)>>1;
      if(L<=m)
        update1(rt<<1,l,m,L,R,k);
      if(R>m)
        update1(rt<<1|1,m+1,r,L,R,k);
      sum[rt]=(sum[rt<<1]+sum[rt<<1|1])%p;
    }
    
    il void update2(int rt,int l,int r,int L,int R,ll k)
    {
      if(L<=l&&R>=r)
        {
          sum[rt]=(sum[rt]+k*(r-l+1))%p;
          add[rt]=(k+add[rt])%p;
          return;
        }
      pushdown(rt,l,r);
      int m=(l+r)>>1;
      if(L<=m)
        update2(rt<<1,l,m,L,R,k);
      if(R>m)
        update2(rt<<1|1,m+1,r,L,R,k);
      sum[rt]=(sum[rt<<1]+sum[rt<<1|1])%p;
    }
    
    il ll query(int rt,int l,int r,int L,int R)
    {
      if(L<=l&&R>=r)
        return sum[rt];
      pushdown(rt,l,r);
      int m=(l+r)>>1;
      ll s=0;
      if(L<=m)
        s=(s+query(rt<<1,l,m,L,R))%p;
      if(R>m)
        s=(s+query(rt<<1|1,m+1,r,L,R))%p;
      return s%p;
    }
    
    int main()
    {
      n=gi(),mm=gi(),p=gl();
      for(int i=1;i<=n;i++)
        a[i]=gl();
      build(1,1,n);
      int x,y,k,r;
      for(int i=1;i<=mm;i++)
        {
          r=gi();
          if(r==1)
    	{
    	  x=gi(),y=gi(),k=gl();
    	  update1(1,1,n,x,y,k);
    	}
          if(r==2)
    	{
    	  x=gi(),y=gi(),k=gl();
    	  update2(1,1,n,x,y,k);
    	}
          if(r==3)
    	{
    	  x=gi(),y=gi();
    	  printf("%lld
    ",query(1,1,n,x,y));
    	}
        }
      return 0;
    }
    
  • 相关阅读:
    学号 20172328 《程序设计与数据结构》第八周学习总结
    172328 结对编程练习_四则运算 第一周 阶段总结
    学号 20172328 《程序设计与数据结构》实验二报告
    20172328《程序设计与数据结构》第七周学习总结
    Educoder
    Educoder
    Educoder
    Educoder
    Educoder
    Educoder
  • 原文地址:https://www.cnblogs.com/gshdyjz/p/9819012.html
Copyright © 2011-2022 走看看