zoukankan      html  css  js  c++  java
  • 线段树(区间乘法和加法模板)

    https://www.luogu.com.cn/problem/P3373

    题意:给出n个数,m次操作,模q , ai , 三种操作:1、l ,  r , k 区间乘k , 2 、 l , r , k , 区间加k , 3 、l ,  r 询问区间和。

    解法:线段树两个标记乘法标记mul和加法标记plu。pushdown下放加法有点特殊,需要自己身上的加法标记乘以父亲的乘法标记再加上

    父亲的加法标记。

    #include<bits/stdc++.h>
    #include <cstdio>
    #include <cstring>
    #include <cmath>
    #include <algorithm>
    #include <iostream>
    #include <string>
    #include <stdio.h>
    #include <queue>
    #include <stack>
    #include <map>
    #include <set>
    #include <string.h>
    #include <vector>
    typedef long long ll ;
    #define int ll
    #define mod 1000000007
    #define gcd __gcd
    #define rep(i , j , n) for(int i = j ; i <= n ; i++)
    #define red(i , n , j)  for(int i = n ; i >= j ; i--)
    #define ME(x , y) memset(x , y , sizeof(x))
    //ll lcm(ll a , ll b){return a*b/gcd(a,b);}
    ll quickpow(ll a , ll b){ll ans=1;while(b){if(b&1)ans=ans*a%mod;b>>=1,a=a*a%mod;}return ans;}
    //int euler1(int x){int ans=x;for(int i=2;i*i<=x;i++)if(x%i==0){ans-=ans/i;while(x%i==0)x/=i;}if(x>1)ans-=ans/x;return ans;}
    //const int N = 1e7+9; int vis[n],prime[n],phi[N];int euler2(int n){ME(vis,true);int len=1;rep(i,2,n){if(vis[i]){prime[len++]=i,phi[i]=i-1;}for(int j=1;j<len&&prime[j]*i<=n;j++){vis[i*prime[j]]=0;if(i%prime[j]==0){phi[i*prime[j]]=phi[i]*prime[j];break;}else{phi[i*prime[j]]=phi[i]*phi[prime[j]];}}}return len}
    #define SC scanf
    #define INF  0x3f3f3f3f
    #define PI acos(-1)
    #define pii pair<int,int>
    #define fi first
    #define se second
    #define lson l,mid,root<<1
    #define rson mid+1,r,root<<1|1
    using namespace std;
    const int N = 1e6+100;
    const int maxn = 1e5+9;
    int n , m , p ;
    int a[maxn];
    struct node{
        int l , r , sum , mul , plu;
    }tree[maxn<<2];
    
    void pushup(int root){
        tree[root].sum = (tree[root<<1].sum + tree[root<<1|1].sum) % p;
    }
    void pushdown(int root){
        int k1 = tree[root].mul , k2 = tree[root].plu;
        tree[root<<1].sum =(k1 * tree[root<<1].sum + k2*(tree[root<<1].r - tree[root<<1].l + 1))% p;
        tree[root<<1|1].sum = (k1 * tree[root<<1|1].sum + k2*(tree[root<<1|1].r - tree[root<<1|1].l + 1))% p;
        tree[root<<1].mul = tree[root<<1].mul * k1 % p ;
        tree[root<<1|1].mul = tree[root<<1|1].mul * k1 % p ;
        tree[root<<1].plu = (tree[root<<1].plu*k1 + k2) % p ;
        tree[root<<1|1].plu = (tree[root<<1|1].plu*k1 + k2) % p ;
        tree[root].mul =1 ;
        tree[root].plu = 0;
    }
    
    void build(int l , int r , int root){
        tree[root].l = l , tree[root].r = r  , tree[root].mul = 1 , tree[root].plu = 0;
        if(l == r){
            tree[root].sum = a[l] % p ;
            return ;
        }
        int mid = (l + r) >> 1;
        build(lson);
        build(rson);
        pushup(root);
    }
    void update_mul(int l , int r , int root , int val){
        if(tree[root].l >= l && tree[root].r <= r){
            tree[root].sum = tree[root].sum * val % p;
            tree[root].mul = tree[root].mul * val % p;
            tree[root].plu = tree[root].plu * val % p;
            return ;
        }
        pushdown(root);
        int mid = (tree[root].l + tree[root].r) >> 1;
        if(mid >= l) update_mul(l , r , root<<1 , val);
        if(mid < r) update_mul(l , r , root<<1|1 , val);
        pushup(root);
    }
    void update_plu(int l , int r , int root , int val){
        if(tree[root].l >= l && tree[root].r <= r){
            tree[root].sum = (tree[root].sum + (tree[root].r - tree[root].l + 1) * val) % p ;
            tree[root].plu = (tree[root].plu + val) % p ;
            return ;
        }
        pushdown(root);
        int mid = (tree[root].l + tree[root].r) >> 1 ;
        if(mid >= l) update_plu(l , r , root<<1 , val);
        if(mid < r) update_plu(l , r , root<<1|1 , val);
        pushup(root);
    }
    
    int query(int l , int r , int root){
        int sum = 0 ;
        if(tree[root].l >= l && tree[root].r <= r){
            return tree[root].sum % p ;
        }
        pushdown(root);
        int mid = (tree[root].l + tree[root].r) >> 1 ;
        if(mid >= l) sum = (sum + query(l , r , root<<1)) % p;
        if(mid < r) sum = (sum + query(l , r , root<<1|1)) % p;
        return sum % p ;
    }
    
    void solve(){
    
        scanf("%lld%lld%lld" , &n , &m , &p);
        rep(i , 1 , n) scanf("%lld" , &a[i]);
        build(1 , n , 1);
        rep(i , 1 , m){
            int t , l , r , val;
            scanf("%lld%lld%lld" , &t , &l , &r);
            if(t == 1){
                scanf("%lld" , &val);
                val %= p;
                update_mul(l , r , 1 , val);
            }else if(t == 2){
                scanf("%lld" , &val);
                val %= p ;
                update_plu(l , r , 1 , val);
            }
            else{
                cout << query(l , r , 1) << endl;
            }
        }
    }
    
    signed main()
    {
        //ios::sync_with_stdio(false);
        //cin.tie(0); cout.tie(0);
        //int t ;
        //cin >> t ;
        //while(t--){
            solve();
        //}
    }
    
  • 相关阅读:
    UrlRewriter
    PortalBasic Web 应用开发框架
    首页静态化和定时执行
    细谈在C#中读写Excel系列文章之四
    Ubuntu安装SVN及SvnManager
    Select语句导致瓶颈
    策略模式
    抽象工厂
    递归优化之快速排序
    从安装git开始,一步一步把程序发布到AppHarbor
  • 原文地址:https://www.cnblogs.com/nonames/p/12489347.html
Copyright © 2011-2022 走看看