$10$進数字列で累乗とか書きたくないからpythonに逃げるもTLEしc++に戻され、行列累乗書くの面倒なので$F_n = F_{\lfloor \frac{n}{2} \rfloor} F_{\lceil \frac{n}{2} \rceil} + F_{\lfloor \frac{n}{2} \rfloor - 1} F_{\lceil \frac{n}{2} \rceil - 1}$1をやってみたら(1ケース5秒なら通ってたんだろうけど)速度が足りなくて、結局まじめに全部書くはめになった。

No.147 試験監督(2)

解法

${\rm ans} = \Pi_i {\rm fib}(C_i)^D_i$。繰り返し二乗法。$O(N (\log C + \log D))$。

机ごとに座り方は区別するので、それぞれの種類の机ひとつに関して座り方が何通りか求め、これをその個数乗すればよい。 座り方が何通りかはfibonacci数になっている。実験する、あるいは漸化式を立てれば見える。

実装

TLEが厳しい。

#include <iostream>
#include <vector>
#define repeat(i,n) for (int i = 0; (i) < (n); ++(i))
#define repeat_reverse(i,n) for (int i = (n)-1; (i) >= 0; --(i))
typedef long long ll;
using namespace std;
const int mod = 1e9+7;
vector<vector<ll> > operator * (vector<vector<ll> > const & p, vector<vector<ll> > const & q) {
    int n = p.size();
    vector<vector<ll> > r(n, vector<ll>(n));
    repeat (y,n) {
        repeat (z,n) {
            repeat (x,n) {
                r[y][x] += p[y][z] * q[z][x] % mod;
                r[y][x] %= mod;
            }
        }
    }
    return r;
}
ll fib(ll n) {
    vector<vector<ll> > f(2, vector<ll>(2));
    vector<vector<ll> > e(2, vector<ll>(2));
    f[0][0] = f[1][1] = 1;
    e[0][0] = e[0][1] = e[1][0] = 1;
    for (ll i = 1; i <= n; i <<= 1) {
        if (n & i) f = f * e;
        e = e * e;
    }
    return f[0][0];
}
int powi(int x, string const & d) {
    ll y = 1;
    int n = d.size();
    repeat_reverse (i,n) {
        int c = d[i] - '0';
        ll z = 1;
        repeat (j,10) {
            if (j == c) y = y * z % mod;
            z = z * x % mod;
        }
        x = z;
    }
    return y;
}
int main() {
    int n; cin >> n;
    ll ans = 1;
    repeat (i,n) {
        ll c; string d; cin >> c >> d;
        ans = ans * powi(fib(c+1), d) % mod;
    }
    cout << ans << endl;
    return 0;
}

Yukicoder No.147 試験監督(2)