2018焦作网络赛 - Poor God Water 一道水题的教训
本题算是签到题,但由于赛中花费了过多的时间去滴吧格,造成了不必要的浪费以及智商掉线,所以有必要记录一下坑点
题意:方格从1到n,每一格mjl可以选择吃鱼/巧克力/鸡腿,求走到n格时满足
1.每三格不可重复同一种食物
2.每三格均不同食物时中间格子不可吃巧克力
3.每三格前后两格不可同时吃巧克力
以上三个条件的方案数,n<1e10
太长不看版:打表+快速幂AC
长篇吐槽版
很显然的,设\(dp[n][i][j][k]\),走到第\(n\)格时第\(n-2\)格的食物是\(i\),第\(n-1\)的食物是\(j\),第\(n\)的食物是\(k\)的方案数
然后一波转移套上矩阵快速幂,应该没问题?
那么首先应该搞出所有转移,emmm好像方程很多,手写花了一定时间发现写歪了(傻逼做法1
于是写了个程序暴力转移
rep(i,1,3)rep(j,1,3)rep(k,1,3){
printf("dp[i][%d][%d][%d]=",i,j,k);
bool flag=0;
if(i==j&&j==k&&i==k) flag=1;
if(j==2&&i!=2&&k!=2&&i!=k) flag=1;
if(i==2&&k==2) flag=1;
if(flag){
cout<<0<<endl;
continue;
}
flag=0;
int jj=i,kk=j;
rep(ii,1,3){
if(ii==jj&&jj==kk&&ii==kk)continue;
if(jj==2&&ii!=2&&kk!=2&&ii!=kk)continue;
if(ii==2&&kk==2)continue;
printf("dp[i-1][%d][%d][%d]+",ii,jj,kk);
flag=1;
}
cout<<endl;
}
于是得到
dp[i][1][1][1]=0
dp[i][1][1][2]=dp[i-1][2][1][1]+dp[i-1][3][1][1];
dp[i][1][1][3]=dp[i-1][2][1][1]+dp[i-1][3][1][1];
dp[i][1][2][1]=dp[i-1][1][1][2]+dp[i-1][3][1][2];
dp[i][1][2][2]=dp[i-1][1][1][2]+dp[i-1][3][1][2];
dp[i][1][2][3]=0;
dp[i][1][3][1]=dp[i-1][1][1][3]+dp[i-1][2][1][3]+dp[i-1][3][1][3];
dp[i][1][3][2]=dp[i-1][1][1][3]+dp[i-1][2][1][3]+dp[i-1][3][1][3];
dp[i][1][3][3]=dp[i-1][1][1][3]+dp[i-1][2][1][3]+dp[i-1][3][1][3];
dp[i][2][1][1]=dp[i-1][1][2][1]+dp[i-1][2][2][1];
dp[i][2][1][2]=0;
dp[i][2][1][3]=dp[i-1][1][2][1]+dp[i-1][2][2][1];
dp[i][2][2][1]=dp[i-1][1][2][2]+dp[i-1][3][2][2];
dp[i][2][2][2]=0;
dp[i][2][2][3]=dp[i-1][1][2][2]+dp[i-1][3][2][2];
dp[i][2][3][1]=dp[i-1][2][2][3]+dp[i-1][3][2][3];
dp[i][2][3][2]=0;
dp[i][2][3][3]=dp[i-1][2][2][3]+dp[i-1][3][2][3];
dp[i][3][1][1]=dp[i-1][1][3][1]+dp[i-1][2][3][1]+dp[i-1][3][3][1];
dp[i][3][1][2]=dp[i-1][1][3][1]+dp[i-1][2][3][1]+dp[i-1][3][3][1];
dp[i][3][1][3]=dp[i-1][1][3][1]+dp[i-1][2][3][1]+dp[i-1][3][3][1];
dp[i][3][2][1]=0;
dp[i][3][2][2]=dp[i-1][1][3][2]+dp[i-1][3][3][2];
dp[i][3][2][3]=dp[i-1][1][3][2]+dp[i-1][3][3][2];
dp[i][3][3][1]=dp[i-1][1][3][3]+dp[i-1][2][3][3];
dp[i][3][3][2]=dp[i-1][1][3][3]+dp[i-1][2][3][3];
dp[i][3][3][3]=0;
先试试样例,卧槽怎么全是0的
发现忘了前3项特判初始化(傻逼做法2
赶紧加上去
rep(i,1,3)dp[1][0][0][i]=dp[0][0][0][0];
rep(i,1,3)rep(j,1,3)dp[2][0][i][j]+=dp[1][0][0][i];
rep(i,1,3)rep(j,1,3)rep(k,1,3){
if(i==j&&j==k&&i==k)continue;
if(j==2&&i!=2&&k!=2&&i!=k)continue;
if(i==2&&k==2)continue;
dp[3][i][j][k]+=dp[2][0][i][j];
}
搞完后怒上27*27的矩阵,然而TLE(傻逼做法3
再套了个费马小定理,TLE(hhh
dalao指点了不如试试BM?没办法只好上了
得到f[n]=2f[n-1]-f[n-2]+3f[n-3]+2f[n-4]
再怒上一波快速幂,发现答案歪了..挂机10min
最后偶然发现必须后几项开始推才是正确的(傻逼做法4
推完后改代码发现又递推不上去了,两个人同时写都歪了
再次发现递推矩阵的上三角写歪了一位...(傻逼做法5
对了,挪板子的时候发现MOD写成了1e9+9,幸好被队友发现了hhh(傻逼做法6
最后奉上代码
#include<bits/stdc++.h>
#define rep(i,j,k) for(int i=j;i<=k;i++)
#define rrep(i,j,k) for(int i=j;i>=k;i--)
#define println(a) printf("%lld\n",(ll)(a))
#define printbk(a) printf("%lld ",(ll)(a))
using namespace std;
typedef long long ll;
const int maxn = 100;
const int MAXN = maxn;
const ll MOD = 1e9+7;
inline ll mod(ll a){return a%MOD;}
ll read(){
ll x=0,f=1;register char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
return x*f;
}
ll b[5][5]={
{0,0,0,0,0},
{0,2,-1,3,2},
{0,1,0,0,0},
{0,0,1,0,0},
{0,0,0,1,0},
};
ll c[]={0,2956,1286,560,244};
struct Mat{
ll m[5][5],r,c;
void node(int rr,int cc,bool unit=0){
r=rr;c=cc;
memset(m,0,sizeof m);
if(unit) rep(i,1,4) m[i][i]=1;
}
void print(){
rep(i,1,4){
rep(j,1,4){
cout<<m[i][j]<<" ";
}
cout<<endl;
}
}
};
Mat operator * (Mat a,Mat b){
Mat ans;ans.node(a.r,b.c);
rep(i,1,4){
rep(j,1,4){
int t=4;
rep(k,1,t){
ans.m[i][j]+=mod(a.m[i][k]*b.m[k][j]);
ans.m[i][j]=mod(ans.m[i][j]);
}
}
}
return ans;
}
Mat qmod(Mat a,ll n){
Mat ans;ans.node(4,4,1);
while(n){
if(n&1) ans=ans*a;
n>>=1;
a=a*a;
}
return ans;
}
ll ans[]={0,3,9,20,46,106,244,560,1286,2956};
int main(){
int T=read();
while(T--){
Mat base,base2;
base.node(4,4); base2.node(4,4);
rep(i,1,4) rep(j,1,4) base.m[i][j]=b[i][j];
rep(i,1,4) base2.m[i][1]=c[i];
ll n=read();
if(n<=9){
println(ans[n]);
continue;
}
base=qmod(base,n-9);
Mat res=base*base2;
println((res.m[1][1]+MOD)%MOD);
}
return 0;
}
感觉这回比赛得背锅啊orz
