Facebook
From demine, 2 Weeks ago, written in C++.
Embed
Download Paste or View Raw
Hits: 96
  1. #include <iostream>
  2. #include <algorithm>
  3. #include <vector>
  4. using namespace std;
  5. int v[250005],gen[250005];
  6. int t,n,a,b;
  7. int main()
  8. {
  9.     cin>>t;
  10.     for(int i=1;i<=t;i++)
  11.     {
  12.         cin>>n>>c>>d;
  13.         int num=0;
  14.         for(int j=1;j<=n*n;j++)
  15.         {
  16.             cin>>v[j];
  17.         }
  18.         sort(v+1,v+n*n+1);
  19.         gen[1]=v[1];
  20.         for(int j=2;j<=n;j++)
  21.         {
  22.             gen[j]=gen[j-1]+c;
  23.         }
  24.         for(int j=2;j<=n;j++)
  25.         {
  26.             for(int k=1;k<=n;k++)
  27.             {
  28.                 gen[k+j*n-1]=gen[(j-1)*n+k-1]+d;
  29.             }
  30.         }
  31.         sort (gen+1,gen+n*n+1);
  32.         bool ok=1;
  33.         for(int i=1;i<=n*n;i++)
  34.         {
  35.             if(gen[i]!=v[i])
  36.             {
  37.                 ok=0;
  38.                 break;
  39.             }
  40.         }
  41.         if(ok==1)
  42.         {
  43.             cout<<"yes"<<"\n";
  44.         }
  45.         else
  46.         {
  47.             cout<<"no"<<"\n";
  48.         }
  49.        
  50.     }
  51. }
  52.