牛牛的粉丝【循环矩阵】

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


比赛的时候用了一个O(N^{3} log(N))的做法来解决这个问题,显然是TLE的,显然的,我们没有发现矩阵中隐含的条件。

\begin{bmatrix} p_3 & p_1 & 0 &p_2 \\ p_2 & p_3 & p_1 & 0 \\ 0 & p_2 & p_3 &p_1 \\ p_1 & 0 & p_2 & p_3 \end{bmatrix}当我们把这个矩阵列写下来的时候,会发现以第一行为初始的,后面一行相当于前一行右移一位,每一行都具有这样的信息,所以可以将第i行,第j列第值看作是a[i][j] = a[0][j - i],当然,这里的(j - i)是要取模N的。

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

#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#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
#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>= 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

 

你可能感兴趣的:(数论,思维,矩阵)