计蒜客代码填空九宫格题解

计蒜客代码填空九宫格

题面

将数字 1… 91…9 填入一个 3 ×3 的九宫格中，使得格子中每一横行和的值全部相等，每一竖列和的值全部相等。请你计算有多少种填数字的方案。

题解

网上找到了9层for循环裸暴力，最后想起来可以化为一维数组，利用全排列函数next_permutation来解决。当然也是暴力。

精简了代码量√

#include <iostream>
#include <algorithm>
using namespace std;
//@start: 2020-10-12 20:14:18

int main()
{
    int a[9]={1,2,3,4,5,6,7,8,9};
    int ans=0;
    do{
        if(a[0]+a[1]+a[2]==a[3]+a[4]+a[5]&&a[0]+a[1]+a[2]==a[6]+a[7]+a[8] && a[0]+a[3]+a[6]==a[1]+a[4]+a[7]&&a[0]+a[3]+a[6]==a[2]+a[5]+a[8]) ans++;
    }while(next_permutation(a,a+9));
    cout<<ans;
    return 0;
}

答案

72