牛牛的粉丝【循环矩阵】

题目链接【牛客练习赛68】

---

比赛的时候用了一个的做法来解决这个问题，显然是TLE的，显然的，我们没有发现矩阵中隐含的条件。

当我们把这个矩阵列写下来的时候，会发现以第一行为初始的，后面一行相当于前一行右移一位，每一行都具有这样的信息，所以可以将第i行，第j列第值看作是，当然，这里的是要取模N的。

  所以，我们不妨只用维护一行的信息，就可以完整的表述所有的信息了，时间复杂度就可以降了一个N。

#include <iostream>
#include <cstdio>
#include <cmath>
#include <string>
#include <cstring>
#include <algorithm>
#include <limits>
#include <vector>
#include <stack>
#include <queue>
#include <set>
#include <map>
#include <bitset>
#include <unordered_map>
#include <unordered_set>
#define lowbit(x) ( x&(-x) )
#define pi 3.141592653589793
#define e 2.718281828459045
#define INF 0x3f3f3f3f
#define HalF (l + r)>>1
#define lsn rt<<1
#define rsn rt<<1|1
#define Lson lsn, l, mid
#define Rson rsn, mid+1, r
#define QL Lson, ql, qr
#define QR Rson, ql, qr
#define myself rt, l, r
#define pii pair<int, int>
#define MP(a, b) make_pair(a, b)
using namespace std;
typedef unsigned long long ull;
typedef unsigned int uit;
typedef long long ll;
const int mod = 998244353;
const int maxN = 500;
int N, S, A, B, C;
ll K, p1, p2, p3;
ll fast_mi(ll a, ll b = mod - 2)
{
    ll ans = 1;
    while(b)
    {
        if(b & 1) ans = ans * a % mod;
        a = a * a % mod;
        b >>= 1;
    }
    return ans;
}
ll x[maxN], y[maxN], Bas[maxN], dp[maxN], ans[maxN];
inline int _ID(int ith) { return ith < 0 ? ith + N : (ith >= N ? ith - N : ith); }
void Mul(ll *a, ll *b, ll *m)
{
    memset(y, 0, sizeof(ll) * N);
    for(int i=0; i<N; i++) for(int j=0; j<N; j++) y[j] = (y[j] + a[i] * b[_ID(j - i)]) % mod;
    for(int i=0; i<N; i++) m[i] = y[i];
}
void Fast_Matrice(ll b)
{
    while(b)
    {
        if(b & 1LL) Mul(Bas, x, Bas);
        Mul(x, x, x);
        b >>= 1LL;
    }
}
int main()
{
    scanf("%d%lld", &N, &K);
    scanf("%d%d%d", &A, &B, &C);
    S = A + B + C;
    ll inv_s = fast_mi(S);
    p1 = A * inv_s % mod;
    p2 = B * inv_s % mod;
    p3 = C * inv_s % mod;
    x[0] = p3; x[1] = p1; x[N - 1] = p2;
    Bas[0] = 1;
    Fast_Matrice(K);
    for(int i=0; i<N; i++) scanf("%lld", &dp[i]);
    for(int i=0; i<N; i++)
    {
        ll tmp = 0;
        for(int j=0; j<N; j++)
        {
            tmp = (tmp + dp[j] * Bas[_ID(i - j)]) % mod;
        }
        ans[i] = tmp;
    }
    for(int i=0; i<N; i++) printf("%lld%c", ans[i], i == N - 1 ? '\n' : ' ');
    return 0;
}