1436B - Prime Square

• Quiz
  • https://codeforces.com/contest/1436/problem/B
• AC
  • https://codeforces.com/contest/1436/submission/96922880
• My editorial
  • it is different from official editorial, so I wrote.
  • My solution is shown in the image below

void solve(){
  ll N;
  cin>>N;
  if(N==2){
    p("1 1");
    p("1 1");
    return;
  }
  VI A(N);
  rep(i,3)A[i]=1;
  rep(i,N){
    print_vector(A);
    rotate(A.begin(), A.begin()+1, A.end());
  }
}

int main(){
    cin.tie(0);
    ios::sync_with_stdio(false);

    // input
    ll N; 
    cin>>N;
    
    while(N--)solve();
    
    return 0;
}