CF - 718C 线段树+矩阵

题意：

给出n个数a1到an，还有m次操作，操作分为两种类型，一种给下标属于[l,r]的所有ai+x，另一种操作是询问F(a[l])+F(a[l+1])+...+F(a[r]),其中F(i)表示斐波那契数列的第i项。

思路：

求斐波那契数列通常会想到矩阵。可以发现：

根据上面的递推式，可以用线段树维护这一段区间[l,r]中的F(a[i])的和以及F(a[i]-1)的和。对于询问操作直接就可以输出结果，对于修改操作，则可以利用上式直接乘上变化矩阵Am的x次幂。
这道题要注意两点，为了降低复杂度，lazy标记必须保存矩阵，而不能只存幂次，另外，在更新操作的时候update函数的add也要直接先把矩阵算出来当作参数传递，而不能在找到时候快速幂。

代码：

#include <bits/stdc++.h>
#define lson l, m, rt << 1
#define rson m + 1, r, rt << 1 | 1
using namespace std;
typedef long long ll;
const ll MOD = 1e9 + 7;
const int MAXN = 1e5 + 10;

struct node {
    ll x, y;
}sum[MAXN << 2];

struct mat{
    ll p[3][3], sz;
    mat operator + (const mat &x) {
        mat tp; tp.sz = sz;
        for (int i = 1; i <= sz; i++)
            for (int j = 1; j <= sz; j++)
                tp.p[i][j] = (p[i][j] + x.p[i][j]) % MOD;
        return tp;
    }
    mat operator * (const mat &x) {
        mat tp; tp.sz = sz;
        for (int i = 1; i <= sz; i++)
            for (int j = 1; j <= sz; j++) {
                tp.p[i][j] = 0;
                for (int k = 1; k <= sz; k++)
                    tp.p[i][j] = (tp.p[i][j] + p[i][k] * x.p[k][j] % MOD) % MOD;
            }
        return tp;
    }
}unit, Am;

ll a[MAXN];
mat lazy[MAXN << 2];

mat pow_mod(mat a, ll n) {
    mat res = unit;
    while (n) {
        if (n & 1) res = res * a;
        a = a * a;
        n >>= 1;
    }
    return res;
}

void init() {
    memset(&unit, 0, sizeof(unit));
    memset(&Am, 0, sizeof(Am));
    unit.sz = Am.sz = 2;
    for (int i = 1; i <= 2; i++) unit.p[i][i] = 1;
    Am.p[1][1] = Am.p[1][2] = Am.p[2][1] = 1;
}

bool Is_unit(mat x) {
    if (x.p[1][1] || x.p[2][2]) return false;
    return true;
}

void push_up(int rt) {
    sum[rt].x = (sum[rt << 1].x + sum[rt << 1 | 1].x) % MOD;
    sum[rt].y = (sum[rt << 1].y + sum[rt << 1 | 1].y) % MOD;
}

void push_down(int rt) {
    ll x, y;
    if (!Is_unit(lazy[rt])) {
        lazy[rt << 1] = lazy[rt << 1] * lazy[rt];
        lazy[rt << 1 | 1] = lazy[rt << 1 | 1] * lazy[rt];
        x = (sum[rt << 1].x * lazy[rt].p[1][1] % MOD + sum[rt << 1].y * lazy[rt].p[1][2] % MOD) % MOD;
        y = (sum[rt << 1].x * lazy[rt].p[2][1] % MOD + sum[rt << 1].y * lazy[rt].p[2][2] % MOD) % MOD;
        sum[rt << 1] = (node){x, y};
        x = (sum[rt << 1 | 1].x * lazy[rt].p[1][1] % MOD + sum[rt << 1 | 1].y * lazy[rt].p[1][2] % MOD) % MOD;
        y = (sum[rt << 1 | 1].x * lazy[rt].p[2][1] % MOD + sum[rt << 1 | 1].y * lazy[rt].p[2][2] % MOD) % MOD;
        sum[rt << 1 | 1] = (node){x, y};
        lazy[rt] = unit;
    }
}

void build(int l, int r, int rt) {
    lazy[rt] = unit;
    if (l == r) {
        if (a[l] == 1) sum[rt] = (node) {1, 0};
        else if (a[l] == 2) sum[rt] = (node) {1, 1};
        else {
            mat tmp = pow_mod(Am, a[l] - 2);
            sum[rt].x = (1 * tmp.p[1][1] + 1 * tmp.p[1][2]) % MOD;
            sum[rt].y = (1 * tmp.p[2][1] + 1 * tmp.p[2][2]) % MOD;
        }
        return;
    }
    int m = (l + r) >> 1;
    build(lson);
    build(rson);
    push_up(rt);
}

ll query(int L, int R, int l, int r, int rt) {
    if (L <= l && r <= R) return sum[rt].x;
    push_down(rt);
    int m = (l + r) >> 1;
    ll res = 0;
    if (L <= m) res = (res + query(L, R, lson)) % MOD;
    if (R > m) res = (res + query(L, R, rson)) % MOD;
    return res;
}

void update(int L, int R, mat add, int l, int r, int rt) {
    if (L <= l && r <= R) {
        lazy[rt] = lazy[rt] * add;
        ll x = (sum[rt].x * add.p[1][1] % MOD + sum[rt].y * add.p[1][2] % MOD) % MOD;
        ll y = (sum[rt].x * add.p[2][1] % MOD + sum[rt].y * add.p[2][2] % MOD) % MOD;
        sum[rt] = (node){x, y};
        return;
    }
    push_down(rt);
    int m = (l + r) >> 1;
    if (L <= m) update(L, R, add, lson);
    if (R > m) update(L, R, add, rson);
    push_up(rt);
}

int main() {
    //freopen("in", "r", stdin);
    init();
    int n, m;
    scanf("%d%d", &n, &m);
    for (int i = 1; i <= n; i++)
        scanf("%I64d", &a[i]);
    build(1, n, 1);
    while (m--) {
        int op, l, r;
        ll x;
        scanf("%d", &op);
        if (op == 1) {
            scanf("%d%d%I64d", &l, &r, &x);
            mat add = pow_mod(Am, x);
            update(l, r, add, 1, n, 1);
        }
        else {
            scanf("%d%d", &l, &r);
            printf("%I64d\n", query(l, r, 1, n, 1));
        }
    }
    return 0;
}