정사각 배열 대각선 순서로 접근

정사각 배열에서 대각선에 특징이 있습니다.

 

오른쪽 아래로 향하는 대각선의 각 element 들의 x - y 또는 y - x 값이 같습니다.

오른쪽 위로 향하는 대각선은 각 element 들의 x + y 값이 같습니다.

임의의 element 는 이 두 대각선의 교점으로도 해석할 수 있습니다.

즉,

(오른쪽 아래 대각선 값) = x - y (또는 y - x) 

(오른쪽 위 대각선 값) = x + y 

이 두 값이 주어지면 유일한 element의 (y, x) 위치를 알아내 접근할 수 있습니다. 

#include <iostream>
using namespace std;
int mp[101][101];
int n;
 
int main() {
    cin >> n;
    
    int cnt = 1;
    for (int k = 0; k <= 2 * n - 2; k++) {
        int lb, ub;
        if (n <= k) lb = -(2 * n - 2 - k);
        else lb = -k;
        ub = -lb;
        for (int diff = lb; diff <= ub; diff+=2) {
            int x = (k + diff) >> 1, y = k - x;
            mp[y][x] = cnt++;
        }
    }
 
 
    for (int i = 0; i < n; i++) {
        for (int j = 0; j < n; j++)
            cout << mp[i][j] << ' ';
        cout << endl;
    }
 
 
    return 0;
}

 

 

실행 결과