topcoder[SRM 377]外星语言

【题目描述】

一个国际科研小组昨天发现了一张奇奇怪怪的纸片。他们相信它大约有一百万年的历史。而且，它包含了一些用外星语言写下的文本。下面是对于这种外星语言的所有已知事实：
1. 外星语言的字母表由P个元音和Q个辅音构成。
2. 外星语言的每个单词包含了至多N个元音和至多N个辅音。
3. 在单词中元音总是在辅音之前，即每个单词由一段辅音和后面紧跟着的一段元音构成。
元音或辅音的个数都可以是零。
4. 每个单词至少包含一个字母。
5. 每个单词都可以有重音。外星人将重音标记在字母而非音节上。有三种可能的情况：单词没有重音，单词由一个重音（它可以在任一个字母上），单词有两个重音（一个在元音字母上，一个在辅音字母上）。
6. 两个字母拼写完全相同，但重音不同的单词是不同的。
科学家们希望知道外星语言中单词的总数。他们把这个任务作为练习留给了你。
你的程序被要求返回外星语言中单词的总数模M的值。
【输入格式】

一行，有四个整数：P，Q，N，M
【输出格式】

一行，一个整数：外星语言的单词总数模M的值
【样例输入】

sample1:
1 1 1 9

sample2:
2 3 2 1000

sample3:
1 1 1000000000 1000000000

sample4:
123 456 789 987654321
【样例输出】

sample1:
8

sample2:
577

sample3:
0

sample4:
345494202
【提示】

1<=N,P,Q<=10^9
1<=M<=10^9
【来源】

TopCoder Algorithm SRM 377, Div 1, 1000

题解：

可以看出元音和辅音是相对独立的,可以分别计算再乘法原理
令f[i]为选i个元音的方案,f1[i]为无重音方案,f2[i]为有重音方案
g[i]为f[i]的前缀和
那么有

$$g[i]=g[i-1]+f[i]$$
$$f[i]=f1[i]+f2[i]$$
$$f1[i]=f1[i-1]*P$$
$$f2[i]=f1[i]*i$$
数据范围比较大,考虑构造矩阵
| 1 1 0 0 | | g[i-1] | | g[i] |
| 0 0 2P P | | f[i] | = | f[i+1] |
| 0 0 P 0 | | f1[i] | | f1[i+1] |
| 0 0 P 0 | | f2[i] | | f2[i+1] |
正确性？
$$f[i+1]=f1[i+1]+f2[i+1]$$
$$=f1[i]*P+f1[i+1]*(i+1)$$
$$=f1[i]*P+f1[i]*P*(i+1)$$
$$=f1[i]*P+f1[i]*P*i+f1[i]*P$$
$$=f1[i]*2P+f2[i]*P$$
辅音构造类似
时间复杂度O(logn)

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
int mod;
struct matrix{
    long long a[5][5];
    int I,J;
    matrix(){
        memset(a,0,sizeof(a));
        I=J=0;
    }
    matrix operator *(const matrix &b)const{
        matrix c;
        for(int i=1;i<=I;i++)
            for(int j=1;j<=b.J;j++)
                for(int k=1;k<=J;k++){
                    c.a[i][j]+=a[i][k]*b.a[k][j];
                    c.a[i][j]%=mod;
                }
        c.I=I,c.J=b.J;
        return c;
    }
};
matrix f1,f2;
matrix t1,t2;
matrix d1,d2;
matrix qpow1(int x){
    if(x==1)
        return f1;
    matrix tmp=qpow1(x/2);
    tmp=tmp*tmp;
    if(x&1)
        tmp=tmp*f1;
    return tmp;
}
matrix qpow2(int x){
    if(x==1)
        return f2;
    matrix tmp=qpow2(x/2);
    tmp=tmp*tmp;
    if(x&1)
        tmp=tmp*f2;
    return tmp;
}
int main(){
    freopen("alienlanguage.in","r",stdin);
    freopen("alienlanguage.out","w",stdout);
    int p,q,n;
    scanf("%d %d %d %d",&p,&q,&n,&mod);
    f1.I=f1.J=4;
    f2.I=f2.J=4;
    f1.a[1][1]=f1.a[1][2]=f2.a[1][1]=f2.a[1][2]=1;
    f1.a[2][4]=f1.a[3][3]=f1.a[4][3]=f1.a[4][4]=p;
    f1.a[2][3]=2*p;
    f2.a[2][4]=f2.a[3][3]=f2.a[4][3]=f2.a[4][4]=q;
    f2.a[2][3]=2*q;
    t1.I=t2.I=4;
    t1.J=t2.J=1;
    t1.a[1][1]=0;
    t1.a[2][1]=2*p;
    t1.a[3][1]=p;
    t1.a[4][1]=p;
    t2.a[1][1]=0;
    t2.a[2][1]=2*q;
    t2.a[3][1]=q;
    t2.a[4][1]=q;
    d1=qpow1(n);
    d2=qpow2(n);
    d1=d1*t1;
    d2=d2*t2;
    long long ans=(((d1.a[1][1]*d2.a[1][1])%mod+d1.a[1][1])%mod+d2.a[1][1])%mod;
    printf("%lld\n",ans);
return 0;
}