首页 > 代码库 > CodeForces 718C Sasha and Array

CodeForces 718C Sasha and Array

xianduanshu.

#pragma comment(linker, "/STACK:1024000000,1024000000")#include<cstdio>#include<cstring>#include<cmath>#include<algorithm>#include<vector>#include<map>#include<set>#include<queue>#include<stack>#include<bitset>#include<iostream>using namespace std;typedef long long LL;const double pi=acos(-1.0),eps=1e-6;void File(){    freopen("D:\\in.txt","r",stdin);    freopen("D:\\out.txt","w",stdout);}template <class T>inline void read(T &x){    char c=getchar(); x=0;    while(!isdigit(c)) c=getchar();    while(isdigit(c)) {x=x*10+c-0; c=getchar();}}int mod=1e9+7;const int maxn=100010;struct Matrix{    int A[3][3]; int R,C;}s[4*maxn],t[4*maxn];int n,m;Matrix Q,Z,P;Matrix ch(Matrix a,Matrix b){    Matrix c;    int i, j, k;    for (i = 1; i <= a.R; i++)        for (j = 1; j <= b.C; j++){            c.A[i][j]=0;            for (k = 1; k <= a.C; k++){                LL aa=(LL)a.A[i][k],bb=(LL)b.A[k][j];                LL cc=aa*bb%(LL)mod;                int dd=(int)cc;                c.A[i][j] = (c.A[i][j] + dd)%mod;            }        }    c.R=a.R; c.C=b.C;    return c;}Matrix pow(LL p){    Matrix X,Y;    X.R=2; X.C=2; Y.R=2; Y.C=2;    Y.A[1][1]=1; Y.A[1][2]=0;    Y.A[2][1]=0; Y.A[2][2]=1;    X.A[1][1]=0; X.A[1][2]=1;    X.A[2][1]=1; X.A[2][2]=1;    while (p)    {        if (p % 2 == 1) Y = ch(Y,X);        p = p >> 1;        X = ch(X,X);    }    return Y;}void pushUp(int rt){    s[rt].R=2; s[rt].C=2;    s[rt].A[1][1]=(s[2*rt].A[1][1]+s[2*rt+1].A[1][1])%mod;    s[rt].A[1][2]=(s[2*rt].A[1][2]+s[2*rt+1].A[1][2])%mod;    s[rt].A[2][1]=(s[2*rt].A[2][1]+s[2*rt+1].A[2][1])%mod;    s[rt].A[2][2]=(s[2*rt].A[2][2]+s[2*rt+1].A[2][2])%mod;}bool check(Matrix x){    if(x.A[1][1]!=1) return 0;    if(x.A[1][2]!=0) return 0;    if(x.A[2][1]!=0) return 0;    if(x.A[2][2]!=1) return 0;    return 1;}void pushDown(int rt){    if(check(t[rt])==1) return;    s[2*rt]=ch(s[2*rt],t[rt]); t[2*rt]=ch(t[2*rt],t[rt]);    s[2*rt+1]=ch(s[2*rt+1],t[rt]); t[2*rt+1]=ch(t[2*rt+1],t[rt]);    t[rt].A[1][1]=1; t[rt].A[1][2]=0;    t[rt].A[2][1]=0; t[rt].A[2][2]=1;}void build(int l,int r,int rt){    t[rt].A[1][1]=1; t[rt].A[1][2]=0;    t[rt].A[2][1]=0; t[rt].A[2][2]=1;    t[rt].R=2; t[rt].C=2;    if(l==r)    {        LL x; scanf("%lld",&x);        s[rt].R=2; s[rt].C=2;        s[rt].A[1][1]=1; s[rt].A[1][2]=0;        s[rt].A[2][1]=0; s[rt].A[2][2]=1;        s[rt]=ch(s[rt],pow(x-1));        return;    }    int m=(l+r)/2;    build(l,m,2*rt);    build(m+1,r,2*rt+1);    pushUp(rt);}void update(int L,int R,LL x,int l,int r,int rt){    if(L<=l&&r<=R)    {        s[rt]=ch(s[rt],P);        t[rt]=ch(t[rt],P);        return ;    }    int m=(l+r)/2;    pushDown(rt);    if(L<=m) update(L,R,x,l,m,2*rt);    if(R>m) update(L,R,x,m+1,r,2*rt+1);    pushUp(rt);}void add(Matrix a){    Q.A[1][1]=(Q.A[1][1]+a.A[1][1])%mod;    Q.A[1][2]=(Q.A[1][2]+a.A[1][2])%mod;    Q.A[2][1]=(Q.A[2][1]+a.A[2][1])%mod;    Q.A[2][2]=(Q.A[2][2]+a.A[2][2])%mod;}void get(int L,int R,int l,int r,int rt){    if(L<=l&&r<=R) { add(s[rt]); return ; }    int m=(l+r)/2;    pushDown(rt);    if(L<=m) get(L,R,l,m,2*rt);    if(R>m) get(L,R,m+1,r,2*rt+1);    pushUp(rt);}int main(){    scanf("%d%d",&n,&m);    build(1,n,1);    for(int i=1;i<=m;i++)    {        int op; scanf("%d",&op);        int L,R; scanf("%d%d",&L,&R);        if(op==1)        {            LL x; scanf("%lld",&x);            P=pow(x); update(L,R,x,1,n,1);        }        else        {            Q.R=2; Q.C=2; memset(Q.A,0,sizeof Q.A);            get(L,R,1,n,1);            Z.R=1; Z.C=2; Z.A[1][1]=1; Z.A[1][2]=1;            Z=ch(Z,Q);            printf("%d\n",Z.A[1][1]);        }    }    return 0;}

 

CodeForces 718C Sasha and Array