牛客之数独合理吗？

链接：https://ac.nowcoder.com/acm/contest/19306/1025
来源：牛客网

时间限制：C/C++ 1秒，其他语言2秒
空间限制：C/C++ 262144K，其他语言524288K
64bit IO Format: %lld

题目描述

众所周知，数独是一款简单上手（划掉）且极易打发时间的游戏，
fishfloss喜欢玩数独，虽然自己很菜。
数独的具体规则是这样的：
需要根据9×9盘面上的已知数字，推理出所有剩余空格的数字，
并满足每一行、每一列、每一宫（3x3）内的数字均含1-9并不重复

例如 ：

1 2 3 4 5 6 7 8 9
4 5 6 7 8 9 1 2 3
7 8 9 1 2 3 4 5 6
2 3 1 5 6 4 8 9 7
5 6 4 8 9 7 2 3 1
8 9 7 2 3 1 5 6 4
3 1 2 6 4 5 9 7 8
6 4 5 9 7 8 3 1 2
9 7 8 3 1 2 6 4 5
就是一个合法的数独的解
而
1 2 3 4 5 6 7 8 9
4 5 6 7 8 9 1 2 3
7 8 9 1 2 3 4 5 6
2 3 1 5 6 4 8 9 7
5 6 4 8 9 7 2 3 1
8 9 7 2 3 1 5 6 4
3 1 2 6 4 5 9 7 8
6 4 5 9 7 8 3 1 2
9 7 8 3 1 2 6 4 4
则不是
那么下面给出数独盘面，来判断一下它是否是数独的合法解吧

输入描述:

输入共9行，每行9个数，代表数独盘面

输出描述:

一行，如果输入是数独的合法解，则输出YES
反之输出NO

示例1

输入
复制
1 2 3 4 5 6 7 8 9
4 5 6 7 8 9 1 2 3
7 8 9 1 2 3 4 5 6
2 3 1 5 6 4 8 9 7
5 6 4 8 9 7 2 3 1
8 9 7 2 3 1 5 6 4
3 1 2 6 4 5 9 7 8
6 4 5 9 7 8 3 1 2
9 7 8 3 1 2 6 4 5
输出
复制
YES

示例2

输入
复制
1 2 3 4 5 6 7 8 9
4 5 6 7 8 9 1 2 3
7 8 9 1 2 3 4 5 6
2 3 1 5 6 4 8 9 7
5 6 4 8 9 7 2 3 1
8 9 7 2 3 1 5 6 4
3 1 2 6 4 5 9 7 8
6 4 5 9 7 8 3 1 2
9 7 8 3 1 2 6 4 4
输出
复制
NO

说明

对于宫9，行9及列9，显然有重复的4，故不满足合法解

用于学习记录，如果大佬有更好的解法，欢迎大佬指教

代码：

#include 
#include 
#include 
#include 
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
using namespace std;
typedef long long LL;
int a[15][15],f[10];
int check(int i,int j)
{
memset(f,0,sizeof(f));//初始化f

 for(int k1=(i/3)*3+1;k1<=(i/3)*3+3;k1++)//转化求每一个三宫
{
 for(int k2=(j/3)*3+1;k2<=(j/3)*3+3;k2++)
 {
   if(!f[a[k1][k2]])
   {
    f[a[k1][k2]]=1;
   }
   else
   {
    return 1;
   }
 }
}
  return 0;
}
int main()
{
for(int i=1;i<=9;i++)
{
 for(int j=1;j<=9;j++)
 {
  cin>>a[i][j];
 }
}
int flag=0;
//行检查
for(int i=1;i<=9;i++)
{
memset(f,0,sizeof(f));
 for(int j=1;j<=9;j++)
 {
  if(!f[a[i][j]])
  {
   f[a[i][j]]=1; 
  }
  else
  {flag=1;
   break;
  }
 }
 if(flag)
 break;
}
if(!flag){//列检查
for(int i=1;i<=9;i++){
 memset(f,0,sizeof(f));
 for(int j=1;j<=9;j++)
 {
  if(!f[a[j][i]])
  {
   f[a[j][i]]=1;
  }
  else
  {
      flag=1;
   break;
  }
 }
 if(flag)
 break;
}
}
 if(!flag)//三宫检查
 {
  for(int i=1;i<=9;i+=3)
  {
  for(int j=1;j<=9;j+=3)
  {
  flag=check(i,j);
  if(flag)
  {
  break;
  }
  }
  if(flag)
  {
  break;
  }
  }
 }
if(!flag)
 cout<<"YES"<<endl;
else
cout<<"NO"<<endl;
return 0;
}

欢迎大佬指点一二