Binary Strings Gym - 101161G 矩阵快速幂 + 打表

http://codeforces.com/gym/101161/attachments

这题通过打表，可以知道长度是i的时候的合法方案数。

然后得到f[1] = 2, f[2] = 3, f[3] = 5, f[4] = 8......这样的广义fib数列

现在要求f[k] + f[2k] + f[3k] + ...... + f[xk]的总和。

直接做很难做，我不知道f[i * k] = x * f[(i - 1) * k] + y * f[(i - 2) * k]

推不出系数的话，有一个结论就是：fib的任意一项肯定能表示成x * f(i - 1) + y * f(i - 2)，就是两个连续的fib数字，能表示出后面所有的fib数列。知道这样的东西后。

可以知道，比如k = 2

形如f(6) = 2 * f(4) + f(3)、然后f(8) = 2 * f(6) + f(5)

那么就可以构造矩阵了。

sum, f(4)，f(3)   ----->   newSum, f(6), f(5)

同样用f(4)和f(3)把f(5)表示出来即可（注意：肯定能表示）

一般构造了矩阵后，写一个求第k项的函数出来，往往比较有用

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <vector>
#include <queue>
#include <string>
#include <stack>
#include <map>
#include <set>
#include <bitset>
#define X first
#define Y second
#define lson T[rt].l
#define rson T[rt].r
#define clr(u,v); memset(u,v,sizeof(u));
#define in() freopen("data.txt","r",stdin);
#define out() freopen("ans","w",stdout);
#define Clear(Q); while (!Q.empty()) Q.pop();
#define pb push_back

using namespace std;
typedef long long LL;
typedef pair<int, int> pii;
const int maxn = ;
const int MOD = 1e9 + ;
struct Matrix {
    LL a[maxn][maxn];
    int row, col;
} base, a;
struct Matrix mul(struct Matrix a, struct Matrix b, int MOD) {
    struct Matrix c = {};
    c.row = a.row, c.col = b.col;
    for (int i = ; i <= a.row; ++i) {
        for (int j = ; j <= b.col; j++) {
            for (int k = ; k <= b.row; ++k) {
                c.a[i][j] += a.a[i][k] * b.a[k][j];
                c.a[i][j] %= MOD;
            }
        }
    }
    return c;
}
struct Matrix qp(Matrix ans, Matrix base, LL n, LL MOD) {
    while (n) {
        if (n & ) {
            ans = mul(ans, base, MOD);
        }
        n >>= ;
        base = mul(base, base, MOD);
    }
    return ans;
}
int f;
void initBaseFib() {
    base.row = base.col = ;
    base.a[][] = , base.a[][] = ;
    base.a[][] = , base.a[][] = ;
}
LL findK(int a, int b, LL k) {
    if (k == ) return a;
    if (k == ) return b;
    initBaseFib();
    Matrix t;
    t.row = , t.col = ;
    t.a[][] = b, t.a[][] = a;
    t = qp(t, base, k - , MOD);
    return t.a[][];
}
void initBaseSum(int a, int b, int c, int d) {
    base.row = base.col = ;
    base.a[][] = , base.a[][] = , base.a[][] = ;
    base.a[][] = , base.a[][] = a, base.a[][] = b;
    base.a[][] = , base.a[][] = c, base.a[][] = d;
}
int pre[];
int sum[];
LL getpos(LL n, LL k) {
    if (n == ) return ;
    if (n == ) return findK(, , n * k);
    if (k == ) {
        initBaseSum(, , , );
        Matrix t;
        t.row = , t.col = ;
        t.a[][] = , t.a[][] = , t.a[][] = ;
        t = qp(t, base, n - , MOD);
        return t.a[][];
    } else if (k == ) {
        initBaseSum(, , , );
        Matrix t;
        t.row = , t.col = ;
        t.a[][] = , t.a[][] = , t.a[][] = ;
        t = qp(t, base, n - , MOD);
        return t.a[][];
    } else {
        Matrix t;
        t.row = , t.col = ;
        t.a[][] = findK(, , k), t.a[][] = findK(, ,  * k), t.a[][] = findK(, ,  * k - );
        initBaseSum(findK(,,k-), findK(,,k-), findK(,,k-), findK(,,k-));
        t = qp(t, base, n - , MOD);
        return t.a[][];
    }
}
void work() {
    LL be, en, k;
    cin >> be >> en >> k;
    if (k == ) {
        //sumofall fib
        if (en <= ) {
            printf("Case %d: %d\n", ++f, sum[en] - sum[be - ]);
            return;
        }

        initBaseSum(, , , );
        be--;
        a.row = , a.col = ;
        a.a[][] = , a.a[][] = , a.a[][] = ;
        Matrix ret = qp(a, base, en - , MOD);
        LL resEn = ret.a[][];
        LL resBe = ;
        if (be -  >= ) {
            ret = qp(a, base, be - , MOD);
            resBe = ret.a[][];
        }
        printf("Case %d: %d\n", ++f, (resEn - resBe + MOD) % MOD);
        return;
    } else {
        en = en / k;
        if (be % k == ) be = (be - k) / k;
        else be /= k;
//        cout << be << " " << en << endl;
        LL ans = (getpos(en, k) - getpos(be, k) + MOD) % MOD;
        printf("Case %d: %d\n", ++f, ans);
    }
}
int cnt[];
void dfs(int k, int t1, int t2) {
    if (k == t1 || k == t2) {
        cnt[k]++;
        return;
    }
    dfs(k - , t1, t2);
    dfs(k - , t1, t2);
}
int main() {
#ifdef local
    in();
#else
#endif
    pre[] = , pre[] = , pre[] = , pre[] = ;
    sum[] = , sum[] = , sum[] = , sum[] = ;
//    int one = 6, t1 = 4, t2 = 2;
//    dfs(one, t1, t2);
//    cout << cnt[t1] << " " << cnt[t2] << endl;
    int t;
    scanf("%d", &t);
    while (t--) work();
    return ;
}