bzoj2875 [Noi2012]随机数生成器

题目

线性递推式，可以直接推公式，或者直接上矩阵吧，矩阵十分简单。
{a,0}
{1,1}变换矩阵。。

注意要用快（man）速乘法，不然要爆long long

#include<bits/stdc++.h>
#define N 100000
#define LL long long
using namespace std;
struct Tree{
    int l;
    int r;
    LL sum;
    bool flag;
};Tree T[N*4+1];
int A[N+1];
int n,m;
int opt,l,r;
void pushup(int rt)
{
    T[rt].sum=T[rt*2].sum+T[rt*2+1].sum;
    T[rt].flag=T[rt*2].flag&T[rt*2+1].flag;
}
void build(int rt,int l,int r)
{
    T[rt].l=l,T[rt].r=r;
    if(l==r)
    {
        T[rt].sum=A[l];
        T[rt].flag=false;
        return;
    }
    int mid=(l+r)/2;
    build(rt*2,l,mid),build(rt*2+1,mid+1,r);
    pushup(rt);
}
LL qurry(int rt,int l,int r)
{
    if(T[rt].l==l&&T[rt].r==r)
        return T[rt].sum;
    int mid=(T[rt].l+T[rt].r)/2;
    if(r<=mid)return qurry(rt*2,l,r);
    else if(l>mid)return qurry(rt*2+1,l,r);
    else return qurry(rt*2,l,mid)+qurry(rt*2+1,mid+1,r);
}
void modify(int rt,int l,int r)
{
    if(T[rt].l==l&&T[rt].r==r)
    {
        if(T[rt].flag)return;
    }
    if(T[rt].l==T[rt].r&&l==r&&l==T[rt].l)
    {
        LL tmp=sqrt(T[rt].sum);
        if(T[rt].sum==tmp)
        {
            T[rt].flag=true;
            return;
        }
        T[rt].sum=tmp;
        return;
    }
    int mid=(T[rt].l+T[rt].r)/2;
    if(r<=mid)modify(rt*2,l,r);
    else if(l>mid)modify(rt*2+1,l,r);
    else modify(rt*2,l,mid),modify(rt*2+1,mid+1,r);
    pushup(rt);
}
void read(int &x)
{
    x=0;
    char c=getchar();
    while(c<'0'||c>'9')c=getchar();
    while(c>='0'&&c<='9')x=x*10+c-'0',c=getchar();
}
int main()
{
    //freopen("in.txt","r",stdin);
    read(n);
    for(int i=1;i<=n;i++)read(A[i]);
    build(1,1,n);
    read(m);
    for(int i=1;i<=m;i++)
    {
        read(opt),read(l),read(r);
        if(opt==2)
            modify(1,l,r);
        if(opt==1)
            printf("%lld\n",qurry(1,l,r));
    }
    return 0;
}