线段树 + 矩阵 --- ZOJ 3772 Calculate the Function

Calculate the Function

Problem's Link:   http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemCode=3772

---

Mean:

略

analyse:

简单的线段树维护矩阵。

矩阵乘法的结合律（a * b * c == a * (b * c)），注意矩阵乘法不满足分配率（a *b != b * a）。

令 M[x] = [1 A[x]]
              [1     0 ] ，
那么有 [ F[R] ] = M[R] * M[R-1] * ... * M[L+2] * [F[L+1]]
          [F[R-1]]                                                   [ F[L] ]

线段树节点维护上边等式右边前n - 1项的乘值(假设等式右边有n项)。每次询问O（log(n)）。

Time complexity: O(n*logn)

Source code: 

/*
* this code is made by crazyacking
* Verdict: Accepted
* Submission Date: 2015-05-25-20.57
* Time: 0MS
* Memory: 137KB
*/
#include <queue>
#include <cstdio>
#include <set>
#include <string>
#include <stack>
#include <cmath>
#include <climits>
#include <map>
#include <cstdlib>
#include <iostream>
#include <vector>
#include <algorithm>
#include <cstring>
#define  LL long long
#define  ULL unsigned long long
using namespace std;

const int MAXN = ;
const int MOD = ;
struct Mat
{
    long long m[][];
    Mat()
    {
            memset(m, , sizeof(m));
    }
    Mat operator * (const Mat &b)
    {
        Mat temp;
        for(int i = ; i < ; i++)
            for(int j = ; j < ; j++)
                for(int k = ; k < ; k++)
                    temp.m[i][j] = ((m[i][k] * b.m[k][j]) + temp.m[i][j]) % MOD;
        return temp;
    }
};

struct Node
{
    int l, r;
    Mat mat;
};
Node node[MAXN << ];
int a[MAXN];

void Build(int rt, int l, int r)
{
    int m;
    node[rt].l = l;
    node[rt].r = r;
    if(l == r)
    {
        node[rt].mat.m[][] = ;
        node[rt].mat.m[][] = a[l];
        node[rt].mat.m[][] = ;
        node[rt].mat.m[][] = ;
    }
    else
    {
        m = (l + r) >> ;
        Build(rt << , l, m);
        Build(rt <<  | , m + , r);
        node[rt].mat = node[rt <<  | ].mat * node[rt << ].mat;//注意顺序
    }

}
Mat Query(int rt, int ql, int qr)
{
    int m;
    if(ql == node[rt].l && node[rt].r == qr)
        return node[rt].mat;
    else
    {
        m = (node[rt].l + node[rt].r) >> ;
        if(qr <= m)
            return Query(rt << , ql, qr);
        else if(ql > m)
            return Query(rt <<  | , ql, qr);
        else
            return Query(rt <<  | , m + , qr) * Query(rt << , ql, m);//注意顺序
    }
}
int T, n, m, ql, qr;

int main()
{
    scanf("%d", &T);
    while(T--)
    {
        Mat res, f;
        scanf("%d%d",&n, &m);
        for(int i = ; i <= n; i++)
            scanf("%d", &a[i]);
        Build(, , n);
        for(int i = ; i<= m; i++)
        {
            scanf("%d%d", &ql, &qr);
            if(qr - ql >= )
            {
                f.m[][] = a[ql + ];
                f.m[][] = a[ql];
                res = Query(, ql + , qr) * f;
                printf("%d\n", res.m[][]);
            }
            else
                printf("%d\n", a[qr]);
        }
    }
    return ;
}