所有元素为非负整数,且各行各列的元素和都等于 7 的 3×3 方阵称为“吉利矩阵”,因为这样的矩阵一共有 666 种。
本题就请你统计一下,把 7 换成任何一个 [2,9] 区间内的正整数 L,把矩阵阶数换成任何一个 [2,4] 区间内的正整数 N,满足条件“所有元素为非负整数,且各行各列的元素和都等于 L”的 N×N 方阵一共有多少种?
输入格式:
输入在一行中给出 2 个正整数 L 和 N,意义如题面所述。数字间以空格分隔。
输出格式:
在一行中输出满足题目要求条件的方阵的个数。
输入样例:
7 3
输出样例:
666
#include<bits/stdc++.h>
using namespace std;
int l,n;
int main(){
cin>>l>>n;
if(n == 2){
int ans = 0;
for(int i = 0; i <= 9;i++){
if(l-i>=0)
ans++;
}
cout<<ans;
}else if(n == 3){
int a3[10] = {0,0,21,55,120,231,406,666,1035,1540};
cout<<a3[l];
}else{
int a4[10] = {0,0,282,2008,10147,40176,132724,381424,981541,0};
cout<<a4[l];
}
return 0;
}