Ruudukko

#include <bits/stdc++.h>
using namespace std;
 
int main(){
    int size;
    cin >> size;
    int grid[size][size] = {0};
 
    for(int y = 0; y < size; y++){
        for(int x = 0; x < size; x++){
            grid[y][x] = x+1;
        }
    }
 
    for(int y = 1; y < size; y++){
        for(int x = 1; x < size; x++){
            bool swap = false;
 
            for(int yy = y; yy >= 0; yy--){
                if(grid[yy][x] == grid[y][x]){
                    swap = true;
                    break;
                }
            }
 
            if(!swap){
                for(int xx = x; xx >= 0; xx--){
                    if(grid[y][xx] == grid[y][x]){
                        swap = true;
                        break;
                    }
                }
            }
 
            if(swap){
                for(int i = 0; i < size; i++){
 
                    bool canswap = true;
 
                    for(int yy = y; yy >= 0; yy--){
                        if(grid[yy][i] == grid[y][x]){
                            canswap = false;
                            break;
                        }
                    }
 
                    if(canswap){
                        int temp = grid[y][x];
                        grid[y][x] = grid[y][i];
                        grid[y][i] = temp;
                        break;
                    }
 
                }
            }
        }
    }
 
    for(int y = 0; y < size; y++){
        for(int x = 0; x < size; x++){
            cout << grid[y][x] << " ";
        }
        cout << endl;
    }
}

Test 1

Verdict: ACCEPTED

Test 2

Verdict: ACCEPTED

Test 3

Verdict: WRONG ANSWER

Test 4

Verdict: WRONG ANSWER

Test 5

Verdict: WRONG ANSWER

Test 6

Verdict: WRONG ANSWER