蓝桥杯 2020 第一场 第4题：矩阵

题目

第四题：矩阵
题目描述
把1∼2020放在2×1010的矩阵里。要求同一行中右边的比左边大，同一列中下边的比上边的大。一共有多少种方案？
答案很大，你只需要给出方案数除以2020的余数即可。

思路

利用动态规划
dp[i][j]：第一行有 i 个数字，第二行有 j 个数字的方案数量；

代码

#include <iostream>
#include <cmath>
using namespace std;
int dp[1020][1020];
int main()
{
	dp[0][0]=1;
	for(int i=0; i<=1010; i++){
		for(int j=0; j<=i&&j<=1010; j++){//第一行的个数要不小于第二行的
			if(i-1>=j) dp[i][j] += dp[i-1][j];//放在第一行
			if(j) dp[i][j] += dp[i][j-1];//放在第二行
			dp[i][j] %= 2020;//提前取余
		}
	}
	cout<< dp[1010][1010] <<endl;//1010
	return 0;
}

答案

1340

dfs有问题

#include <iostream>

using namespace std;
typedef long long ll;
int a[2+10][1010+10];
int hasht[2020+10];//判断1-2020是否放过了
ll ans = 0;

void dfs(int i, int j){//放第i行，第j列元素
    if(i == 2 && j == 1010){//找到一个方案
        ans = (ans+1)%2020;
        cout << ans << endl;
        return;
    }
    for(int k = 1; k <= 2020; k++){//对于第i行，第j列，尝试1-2020
        if(hasht[k] == 0){
            if(i - 1 >= 1) {
                if(k <= a[i-1][j]) continue;//要比上面的数大
            }
            if(j - 1 >= 1){
                if(k <= a[i][j-1]) continue;//要比左边的数大
            }
            a[i][j] = k;//可以放这个数
            hasht[k] = 1;//标记用过这个数了
            if(j < 1010) dfs(i,j+1);//往右放
            if(i < 2) dfs(i+1,j);//往下放
            hasht[k] = 0;//恢复标记
        }
    }
}

int main()
{
    dfs(1,1);
    cout << ans <<endl;
    return 0;
}