HDU-4868 Arc of Dream
根据题目给出的公式,写出关于Sn,An*Bn,An,Bn,1五项的转移矩阵然后用矩阵快速幂解决#include<iostream>#include<cstdio>#include<cstring>#include<algorithm>#include<cmath&g
·
根据题目给出的公式,写出关于Sn,An*Bn,An,Bn,1五项的转移矩阵
然后用矩阵快速幂解决
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
typedef long long ll;
const ll MOD=1e9+7;
const int N=5;
struct node
{
ll a[10][10];
};
node shu,ans,mp;
//shu是输入的矩阵,ans是所求答案
node matrix(node x,node y)
{
for(int i=1;i<=N;i++)
for(int j=1;j<=N;j++){
mp.a[i][j]=0;
for(int p=1;p<=N;p++)
mp.a[i][j]=(mp.a[i][j]+x.a[i][p]*y.a[p][j]+MOD)%MOD;
//矩阵乘法
}
return mp;
}
void work(ll k)
{//矩阵快速幂
for(int i=1;i<=N;i++)
for(int j=1;j<=N;j++)
ans.a[i][j]=0;
for(int i=1;i<=N;i++) ans.a[i][i]=1;
node t=shu;
while(k){
if(k&1)
ans=matrix(ans,t);
k>>=1;
t=matrix(t,t);
}
}
int main()
{
ll n,a,b,ax,bx,ay,by;
while(~scanf("%lld",&n))
{
scanf("%lld%lld%lld%lld%lld%lld",&a,&ax,&ay,&b,&bx,&by);
if(n==0)
{
printf("%lld\n",0);
continue;
}
if(n==1)
{
printf("%lld\n",a*b%MOD);
continue;
}
ll ret=0;
memset(shu.a,0,sizeof(shu.a));
shu.a[1][2]=shu.a[2][2]=ax*bx%MOD;
shu.a[1][3]=shu.a[2][3]=ax*by%MOD;
shu.a[1][4]=shu.a[2][4]=ay*bx%MOD;
shu.a[1][5]=shu.a[2][5]=ay*by%MOD;
shu.a[3][3]=ax;shu.a[4][4]=bx;
shu.a[3][5]=ay;shu.a[4][5]=by;
shu.a[1][1]=shu.a[5][5]=1;
work(n-1);
ret=(ret+a*b%MOD*(ans.a[1][1]+ans.a[1][2]))%MOD;
ret=(ret+a%MOD*ans.a[1][3])%MOD;
ret=(ret+b%MOD*ans.a[1][4])%MOD;
ret=(ret+ans.a[1][5])%MOD;
printf("%lld\n",ret);
}
return 0;
}
更多推荐
已为社区贡献12条内容
所有评论(0)