【JZOJ4854】【NOIP2016提高A组集训第6场11.3】小澳的坐标系 题目描述 数据范围测试点 解法 代码 启发
小澳者表也,数学者景也,表动则景随矣。
小澳不喜欢数学,可数学却待小澳如初恋,小澳睡觉的时候也不放过。
小澳的梦境中出现了一个平面直角坐标系,自原点,向四方无限延伸。
小澳在坐标系的原点,他可以向上、向左或者向右走。他可以走n步,但不能经过相同的点。
小澳想知道他有多少种走法。
数据范围测试点
1~2
n<=10
测试点3~4
n<=100
测试点5~6
n<=1000
测试点7~8
n<=10^6
测试点9~10
n<=10^9
解法
设f[i]为输入为i时,答案为f[i],设g[i]=f[i+1]-f[i]。
找规律获得g[i]=2*g[i-1]+g[i-2]。
矩阵乘法优化即可。
理性解法:
设f[i]为第i步走左的方案数,g[i]为第i步往右走的方案数,h[i]为第i步往上走的方案数;
显然:
矩阵乘法优化即可。
代码
#include<iostream>
#include<stdio.h>
#include<string.h>
#include<math.h>
#include<algorithm>
#define ll long long
using namespace std;
const char* fin="coordinate.in";
const char* fout="coordinate.out";
const ll inf=0x7fffffff;
const ll maxn=4,mo=1000000007;
ll n,i,j,k;
struct rect{
ll d[maxn][maxn];
rect(){
memset(d,0,sizeof(d));
}
void operator =(const rect &b){
ll i,j,k;
for (i=0;i<maxn;i++) for (j=0;j<maxn;j++) d[i][j]=b.d[i][j];
}
rect operator *(const rect &b){
rect c;
ll i,j,k;
for (i=0;i<maxn;i++)
for (j=0;j<maxn;j++)
for (k=0;k<maxn;k++){
c.d[i][j]=(c.d[i][j]+d[i][k]*b.d[k][j])%mo;
}
return c;
}
}a,b,ans;
rect power(rect a,ll v){
rect c;
bool bz=false;
while (v){
if (v&1){
if (!bz) c=a,bz=true;
else c=c*a;
}
a=a*a;
v>>=1;
}
return c;
}
int main(){
freopen(fin,"r",stdin);
freopen(fout,"w",stdout);
scanf("%lld",&n);
if (n==0) printf("1");
else if (n==1) printf("3");
else if (n==2) printf("7");
else{
a.d[0][0]=7;a.d[0][1]=3;a.d[0][2]=4;a.d[0][3]=2;
b.d[0][0]=b.d[0][1]=b.d[2][3]=b.d[3][0]=b.d[3][2]=1;
b.d[2][0]=b.d[2][2]=2;
a=a*power(b,n-2);
printf("%lld",a.d[0][0]);
}
return 0;
}
启发
观察题目得,当前方案数肯定由上一步的方案数得出。
尝试写出动态规划方程式,然后进行矩阵乘法优化邻近状态转移。