题目链接【牛客练习赛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;
}