思路:
直接回溯枚举每一个位置填的数,二维肯定是不方便的,我们转成一维,下标x从0到n*n-1。二维数组下标从0到n-1,在一维中下标为x的点在二维中对应行是x/n,列是x%n。
每个数最小能填的是0,最大肯定就是l了,时间复杂度的上限是n的2l次幂,4的18大概是1e11这样。
我们直接标记每行sum和每列sum,因为只有当前填的元素只会影响它所在的行和列,所以只要判断它所在行和列是否满足条件就行了。剪枝一下,具体的就是,当前和不能大于l,因为后面还有元素要加,以及,当前和加上后面元素的最大和能够大于等于l。
代码:
#include <iostream>
using namespace std;
int l,n,row[5],col[5],ans;
void backtrack(int x){
if(x==n*n){
for(int i=0;i<n;i++){
if(row[i]!=l) return;
if(col[i]!=l) return;
}
ans++;
return;
}
for(int i=0;i<=l;i++){//能填的数最小是0,最大是l
//剪枝1:当前行或列值不超过l
if(row[x/n]+i>l||col[x%n]+i>l) break;
row[x/n]+=i;//对应行和更新
col[x%n]+=i;//对应列和更新
//剪枝2:加上其他没有填的数(取最大)能达到l
if(row[x/n]+l*(n-1-x%n)>=l&&col[x%n]+l*(n-1-x/n)>=l)
backtrack(x+1);
row[x/n]-=i;//还原现场
col[x%n]-=i;
}
}
int main(){
cin>>l>>n;
backtrack(0);
cout<<ans;
return 0;
}