吉利矩阵(DFS)
所有元素为非负整数,且各行各列的元素和都等于 7 的 3×3 方阵称为 “吉利矩阵”,因为这样的矩阵一共有 666 种。
本题就请你统计一下,各行各列的元素和都等于 5 的 3×3 方阵一共有多少种?
思路:统计方法数,且数据不大,所以可以考虑DFS
细节:if(x==n-1&&sy[y]+i<s) continue;表示如果这是最后一行,那么如果每一列加上i还是不能到达s的话,那就说明不可取,或者也可以写成 if(x==n-1&&s[y]<l) i=s-s[y];这样就省去了一个个遍历了。
其中x表示某一行,y表示某列
#include<iostream>
using namespace std;
#define int long long
const int N=9;
int ans=0;
int s,n;
int sx[N];
int sy[N];
void dfs(int j)
{if(n*n==j){ans+=1;return;}for(int i=0;i<=s;i++){int x,y;x=j/n;y=j%n;if(sx[x]+i>s||sy[y]+i>s) continue;if(x==n-1&&sy[y]+i<s) continue;sx[x]+=i;sy[y]+=i;dfs(j+1);sx[x]-=i;sy[y]-=i;}
}
signed main()
{ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);cin>>s>>n;dfs(0);cout<<ans<<endl;return 0;
}