Code Submission Evaluation System Login

CSES - HIIT Open 2016

HIIT Open 2016

Contest start:2016-05-28 11:00:00
Contest end:2016-05-28 16:00:00

Task list | Submit code | Submissions | Messages | Scoreboard | Statistics


History
2016-05-28 15:43:58
2016-05-28 15:40:41
2016-05-28 15:40:11
2016-05-28 15:38:34
2016-05-28 15:37:28
2016-05-28 15:31:54
2016-05-28 15:29:59
2016-05-28 15:28:54
2016-05-28 15:28:04
2016-05-28 15:25:47
2016-05-28 15:24:55
2016-05-28 15:18:42
2016-05-28 15:16:31
2016-05-28 14:35:59
2016-05-28 13:13:15
Task:Judge correctness
Sender:Barely Div 1
Submission time:2016-05-28 15:43:58
Status:READY
Result:ACCEPTED

Show test data

Compiler report

input/code.cpp: In function 'void printmat(std::vector<std::vector<char> >)':
input/code.cpp:38:31: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     for(int i = 0; i < M.size(); i++){
                               ^
input/code.cpp:39:33: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
       for(int j = 0; j < M.size(); j++){
                                 ^

Code

#include <iostream>
#include <string>
#include <set>
#include <vector>
#include <utility>

using namespace std;

typedef char LL;
/*
vector<LL> matmul(vector<vector<LL> > M, vector<LL> v){
  int n = v.size();
  vector<LL> ans(n);
  for(int i = 0; i < n; i++){
    LL x = 0;
    for(int j = 0; j < n; j++){
      x += M[i][j] * v[j];
    }
    ans[i] = x % 64;
  }
  return ans;
}*/

void matmul2(vector<LL> & ans, vector<vector<LL> > M, vector<LL> v){
  int n = v.size();
  //vector<LL> ans(n);
  for(int i = 0; i < n; i++){
    int64_t x = 0;
    for(int j = 0; j < n; j++){
      x += M[i][j] * (int64_t) v[j];
    }
    ans[i] = x % 64;
  }
  //return ans;
}

void printmat(vector<vector<LL> > M){
    for(int i = 0; i < M.size(); i++){
      for(int j = 0; j < M.size(); j++){
	cout << M[i][j] << " ";
      }
      cout << endl;
    }
}

void read_matrix(vector< vector<LL> > & M, int64_t n){
  for(int i = 0; i < n; i++){
    string s; cin >> s;
    for(int j = 0; j < n; j++){
      int64_t x = 0;
      if(s[j] >= 'A' && s[j] <= 'Z') x = s[j] - 'A';
      if(s[j] >= 'a' && s[j] <= 'z') x = s[j] - 'a' + 26;
      if(s[j] >= '0' && s[j] <= '9') x = s[j] - '0' + 52;
      if(s[j] == '+') x = 62;
      if(s[j] == '/') x = 63;
      M[i][j] = x;
    }
  }  //return M;
}

void solve(){
  int64_t n; cin >> n;
  vector<vector<LL> > A(n, vector<LL>(n));
  vector<vector<LL> > X(n, vector<LL>(n));
  read_matrix(A, n);
  read_matrix(X, n);
  vector<vector<LL> > A_T(n, vector<LL>(n)); // transpose
  for(int i = 0; i < n; i++){
    for(int j = 0; j < n; j++){
      A_T[i][j] = A[j][i];
    }
  }  
  //printmat(A); printmat(A_T); printmat(X);
  
  for(int test = 0; test < 5; test++){
    vector<LL> v;
    for(int i = 0; i < n; i++){
      v.push_back(rand() % 64);
    }
    
    vector<LL> Xv(n), t(n), AA_Tv(n);
    
    matmul2(Xv, X,v);
    matmul2(t, A_T,v);
    matmul2(AA_Tv, A, t);
    
    //vector<LL> Xv(n); // DEBUG
    //vector<LL> AA_Tv(n); // DEBUG
    
    bool good = true;
    
    for(int i = 0; i < n; i++){
      if(Xv[i] != AA_Tv[i]) good = false;
    }
    
    if(!good){
      cout << 0 << "\n";
      return;
    }
  }
  
  cout << 1 << "\n";
}

int main(){
  ios_base::sync_with_stdio(0); cin.tie(0);
  cin.tie(0);
  srand(345348345);
  int64_t t; cin >> t;
  while(t--) solve();
}