112 幻方 magic square M<set>运用集合的唯一性

#include<iostream>
#include<fstream>
#include<sstream>
#include<vector>
#include<string>
#include<stack>
#include<cmath>
#include<cstdlib>
#include<algorithm>
#include<functional>
#include<ctime>
#include<iomanip>
#include<cstring>
#include <cstdio>
#include <map>
#include <list>
#include <set>
#include <queue>
#include <deque>
#include <climits>
using namespace std;
int main()
{
    ifstream cin("atest.txt");
    int matrix[10][10];
    set<int> s;
    int n;
    int line[10];
    int column[9];
    while(cin>>n)
    {
        if(n==0)  break;
        s.clear();
        for(int i=1;i<=n;i++)
           for(int j=1;j<=n;j++)
           {
               cin>>matrix[i][j];
               s.insert(matrix[i][j]);
           }
        if(s.size()!=n*n)  {cout<<"no !\n";continue;}
        for(int i=1;i<=n;i++)
        {
            line[i]=0;
            column[i]=0;
        }
        int a=0,b=0;/对角线
        for(int i=1;i<=n;i++)
           for(int j=1;j<=n;j++)
           {
               line[i]+=matrix[i][j];
               column[j]+=matrix[i][j];
               if(i==j) a+=matrix[i][j];
               if(i+j==n+1) b+=matrix[i][j];
           }
        s.clear();
        for(int i=1;i<=n;i++)
        {
            s.insert(line[i]);
            s.insert(column[i]);
        }
         s.insert(a);
          s.insert(b);
        if(s.size()!=1)  cout<<"no !\n";
        else   cout<<"yes! \n";

    }

}