简单变相
最优题解
可以发现,每次牛牛所在格子的列数必定+1,则在当前格子的方案数只与上一格3个格子的方案数有关。
所以用表示从(1,1)到(i,j)的方案数,如果(i,j)有障碍物那么方案数就是0,否则等于走一步能到达它的格子的方案数的累加。
因为是按列推进的,所以为了cache友好,把第一维作为列,第二维作为行
时间和空间复杂度都是
int dp[100005][3];//dp(i,j)表示从(1,1)到(j+1,i)的方案数。
int vis[100005][3];
int solve(int n, int m, vector<int> &x, vector<int> &y){
assert(x.size() == m && y.size() == m);
memset(dp, 0, sizeof dp); memset(vis, 0, sizeof vis);
for(int i = 0; i < m; ++i) vis[y[i]][x[i]-1] = 1;
dp[1][0] = 1; int mod = 1e9 + 7;
for(int i = 2; i <= n; ++i){
if(!vis[i][0]) dp[i][0] = (dp[i-1][0] + dp[i-1][1])%mod;
else dp[i][0] = 0;
if(!vis[i][1]) dp[i][1] = ((dp[i-1][0] + dp[i-1][1])%mod + dp[i-1][2])%mod;
else dp[i][1] = 0;
if(!vis[i][2]) dp[i][2] = (dp[i-1][1] + dp[i-1][2])%mod;
else dp[i][2] = 0;
}
assert(dp[n][2] < mod && dp[n][2] >= 0);
return dp[n][2];
}
曼迪匹艾公司福利 125人发布