4499: 线性函数 线段树

来源:互联网 发布:男生美图软件 编辑:程序博客网 时间:2024/05/21 19:24

维护处每个区间内的数的乘积,可以发现将两个区间合并的时候为左边的和乘右边的乘积加右边的和,线段树维护。

#include<iostream>#include<cstdio>using namespace std;const int mod=1e9+7;const int N=200005;int n,m,K[N],b[N];struct node{    int l,r;    long long mul,sum;}t[N<<2];inline int read(){    int a=0,f=1; char c=getchar();    while (c<'0'||c>'9') {if (c=='-') f=-1; c=getchar();}    while (c>='0'&&c<='9') {a=a*10+c-'0'; c=getchar();}    return a*f;}inline node Unite(node x,node y){    node ans;    ans.l=x.l; ans.r=y.r;    ans.mul=x.mul*y.mul%mod;    ans.sum=(x.sum*y.mul%mod+y.sum)%mod;    return ans;}inline void pushup(int k){    t[k]=Unite(t[k<<1],t[k<<1|1]);}void build(int k,int x,int y){    if (x==y)    {        t[k].l=t[k].r=x;        t[k].sum=b[x];        t[k].mul=K[x];        return;    }    int mid=x+y>>1;    build(k<<1,x,mid); build(k<<1|1,mid+1,y);    pushup(k);}void modify(int k,int x,int K,int B){    if (t[k].l==t[k].r)    {        t[k].sum=B;        t[k].mul=K;        return;    }    int mid=t[k].l+t[k].r>>1;    if (x<=mid) modify(k<<1,x,K,B); else modify(k<<1|1,x,K,B);    pushup(k);}node query(int k,int x,int y){    if (t[k].l==x&&t[k].r==y) return t[k];    int mid=t[k].l+t[k].r>>1;    if (y<=mid) return query(k<<1,x,y);    else if (x>mid) return query(k<<1|1,x,y);    else return Unite(query(k<<1,x,mid),query(k<<1|1,mid+1,y));}int main(){    n=read(); m=read();    for (int i=1;i<=n;i++)        K[i]=read(),b[i]=read();    build(1,1,n);    for (int i=1;i<=m;i++)    {        char opt[5];        scanf("%s",opt);        if (opt[0]=='M')        {            int x=read(),k=read(),b=read();            modify(1,x,k,b);        }        else        {            int l=read(),r=read(),x=read();            node ans=query(1,l,r);            printf("%lld\n",(ans.mul*x%mod+ans.sum)%mod);        }    }    return 0;}
0 0
原创粉丝点击