[HDU1757] A Simple Math Problem(매트릭스 신속 멱)

3615 단어 문제풀이dp행렬

제목 설명


전송문
제목의 뜻 f(i)의 정의는 제목의 면을 보세요. 바로 그 추이식입니다.n,Mod를 주고 f(n)를 구하고 답은 Mod에 대한 모형을 취한다.

문제풀이


간단한 행렬 가속 dp 문제.명령하다
A=[fn−1fn−2⋯fn−10]
B=⎡⎣⎢⎢⎢⎢⎢a1a2⋮a1010⋮001⋮0⋯⋯⋱⋯00⋮1⎤⎦⎥⎥⎥⎥⎥
... 해야 한다
A∗Bn - 9이 답이다

코드

#include
#include
#include
using namespace std;
#define LL long long
LL n,Mod;
struct hp{LL a[20][20];}unit,A,m,ans;

inline hp cheng(hp a,hp b)
{
    hp ans;
    memset(ans.a,0,sizeof(ans.a));
    for (int i=1;i<=10;++i)
        for (int j=1;j<=10;++j)
            for (int k=1;k<=10;++k)
                ans.a[i][j]=(ans.a[i][j]+a.a[i][k]*b.a[k][j]%Mod)%Mod;
    return ans;
}
inline hp matrix_fast_pow(hp a,LL p)
{
    hp ans=unit;
    for (;p;p>>=1,a=cheng(a,a))
        if (p&1)
            ans=cheng(ans,a);
    return ans;
}
int main()
{
    for (int i=1;i<=10;++i) unit.a[i][i]=1;
    for (int i=9;i>=0;--i) A.a[1][10-i]=(LL)i;
    for (int i=2;i<=10;++i) m.a[i-1][i]=1;
    while (~scanf("%I64d%I64d",&n,&Mod))
    {
        for (int i=1;i<=10;++i) scanf("%I64d",&m.a[i][1]);
        ans=matrix_fast_pow(m,n-9);
        ans=cheng(A,ans);
        printf("%I64d
"
,ans.a[1][1]); } }

총괄 주의 행렬이 교환율을 만족시키지 못한다.

좋은 웹페이지 즐겨찾기