202012-5 Star Trek

202012-5 Star Trek

80 points method, because N is too big, I don't know how to deal with
it. My method: x , y , zx,y,zx,and ,A line segment tree is opened on the z -axis, and then the line segment tree is applied, interval addition, interval multiplication, and then there is a line segment tree to exchange nodes with each other, which needs to be pushed down first and then exchanged. After the exchange is completed, pushup is required.

//80分做法
#include <stdio.h>
#include <iostream>
#include <queue>
#include <map>
#include <vector>
#include <algorithm>
#include <cstring>
using namespace std;
typedef long long ll;
const ll maxn = 1e6+10;

ll N,M;
const ll mod = 1e9 + 7;
struct node{
    
    
    ll l,r,lson,rson;
    ll add,mul,sum;
}tr[maxn*4];ll idx;
ll rtx,rty,rtz;
struct Seg{
    
    
    void pushup(node &U,node& L,node& R){
    
    
        U.sum = (L.sum + R.sum)%mod;
    }
    ll build(ll l,ll r){
    
    
        tr[++idx] = {
    
    l,r,-1,-1,0,1,0};
        ll id = idx;
        if(l == r){
    
    
            return idx;
        }else{
    
    
            ll mid = (l+r)>>1;
            tr[id].lson = build(l,mid);
            tr[id].rson = build(mid+1,r);
            pushup(tr[id],tr[tr[id].l],tr[tr[id].r]);
        }
        return id;
    }
    void pushdown(node &U,node &L,node &R){
    
    
        L.sum = (L.sum * U.mul%mod + (L.r - L.l + 1) * U.add%mod) %mod;
        R.sum = (R.sum * U.mul%mod + (R.r - R.l + 1) * U.add%mod) %mod;

        L.add = (L.add * U.mul %mod + U.add)%mod;
        R.add = (R.add * U.mul %mod + U.add)%mod;
        L.mul = (L.mul * U.mul)%mod;
        R.mul = (R.mul * U.mul)%mod;
        U.add = 0;
        U.mul = 1;
    }

    void add(ll u,ll l,ll r,ll v){
    
    
        if(l <= tr[u].l && tr[u].r <= r){
    
    
            tr[u].sum = (tr[u].sum + v * (tr[u].r-tr[u].l+1)%mod) %mod;
            tr[u].add = (tr[u].add + v )%mod;
        }else{
    
    
            pushdown(tr[u],tr[tr[u].lson],tr[tr[u].rson]);
            ll mid = (tr[u].l + tr[u].r)>>1;
            if(l <= mid) add(tr[u].lson,l,r,v);
            if(r > mid) add(tr[u].rson,l,r,v);
            pushup(tr[u],tr[tr[u].lson],tr[tr[u].rson]);
        }
    }
    void mul(ll u,ll l,ll r,ll v){
    
    
        if(l<= tr[u].l && tr[u].r <= r){
    
    
            tr[u].sum = (tr[u].sum * v)%mod;
            tr[u].mul = (tr[u].mul * v)%mod;
            tr[u].add = (tr[u].add * v)%mod;
        }else{
    
    
            pushdown(tr[u],tr[tr[u].lson],tr[tr[u].rson]);
            ll mid = (tr[u].l + tr[u].r)>>1;
            if(l<=mid) mul(tr[u].lson,l,r,v);
            if(r>mid) mul(tr[u].rson,l,r,v);
            pushup(tr[u],tr[tr[u].lson],tr[tr[u].rson]);
        }
    }
    node query(ll u,ll l,ll r){
    
    
        if(l <= tr[u].l && tr[u].r <= r){
    
    
            return tr[u];
        }else{
    
    
            pushdown(tr[u],tr[tr[u].lson],tr[tr[u].rson]);
            ll mid = (tr[u].l + tr[u].r)>>1;
            if(r<=mid) return query(tr[u].lson,l,r);
            else if(l>mid) return query(tr[u].rson,l,r);
            else{
    
    
                node ans1 = query(tr[u].lson,l,r);
                node ans2 = query(tr[u].rson,l,r);
                node ans;
                pushup(ans,ans1,ans2);
                return ans;
            }
        }
    }
    void find(vector<ll> &ve,ll u,ll l,ll r){
    
    
        if(l<= tr[u].l && tr[u].r <= r){
    
    
            ve.push_back(u);
        }else{
    
    
            pushdown(tr[u],tr[tr[u].lson],tr[tr[u].rson]);
            ll mid = (tr[u].l+tr[u].r)>>1;
            if(l<=mid) find(ve,tr[u].lson,l,r);
            if(r>mid) find(ve,tr[u].rson,l,r);
            pushup(tr[u],tr[tr[u].lson],tr[tr[u].rson]);
        }
    }
    void change(ll l,ll r){
    
    
        vector<ll> vex,vey,vez;
        find(vex,rtx,l,r);
        find(vey,rty,l,r);
        find(vez,rtz,l,r);
        for(ll i = 0;i<vex.size();i++){
    
    
            ll x = vex[i],y = vey[i],z = vez[i];
            swap(tr[x],tr[z]);
            swap(tr[x],tr[y]);
        }
        find(vex,rtx,l,r);
        find(vey,rty,l,r);
        find(vez,rtz,l,r);
    }
}seg;

int main(){
    
    
    cin>>N>>M;
    rtx = seg.build(1,N);
    rty = seg.build(1,N);
    rtz = seg.build(1,N);
    while(M--){
    
    
        ll op,l,r;scanf("%lld %lld %lld",&op,&l,&r);
        if(op == 1){
    
    
            ll a,b,c;scanf("%lld %lld %lld",&a,&b,&c);
            seg.add(rtx,l,r,a);
            seg.add(rty,l,r,b);
            seg.add(rtz,l,r,c);
        }
        if(op == 2){
    
    
            ll k;scanf("%lld",&k);
            seg.mul(rtx,l,r,k);
            seg.mul(rty,l,r,k);
            seg.mul(rtz,l,r,k);
        }
        if(op == 3){
    
    
            seg.change(l,r);
        }
        if(op == 4){
    
    
            ll sumx = seg.query(rtx,l,r).sum; \\cout<<sumx<<'\n';
            ll sumy = seg.query(rty,l,r).sum; \\cout<<sumy<<'\n';
            ll sumz = seg.query(rtz,l,r).sum; \\cout<<sumz<<'\n';
            printf("%lld\n",(sumx*sumx%mod + sumy*sumy%mod + sumz*sumz%mod)%mod);
        }
    }

    return 0;
}

Guess you like

Origin http://43.154.161.224:23101/article/api/json?id=326267213&siteId=291194637