백준[1913] 달팽이

문제 설명

문제 링크

http://icpc.me/1913

1913번: 달팽이

풀이

구현 문제다.

달팽이처럼 뱅글뱅글 돌면서 순회를 하는 문제다.

구현하는 방식을 알아두자.

코드

#include <cstdio>
#include <algorithm>

int n, k;
int dir[4][2] = {{1, 0}, {0, 1}, {-1, 0}, {0, -1}};
bool visited[1003][1003];
int arr[1003][1003];

bool movable(int r, int c, int d) {
    int nr = r + dir[d][0];
    int nc = c + dir[d][1];
    if (nr < 1 || nr > n || nc < 1 || nc > n) return false;
    if (visited[nr][nc]) return false;
    return true;
}

int main() {
    scanf("%d %d", &n, &k);
    int r = 1, c = 1, d = 0;
    std::pair<int, int> ret;
    for (int cur = n * n; cur; cur--) {
        if (cur == k) ret = {r, c};
        arr[r][c] = cur;
        visited[r][c] = true;
        if (!movable(r, c, d)) d = (d + 1) % 4;
        r += dir[d][0];
        c += dir[d][1];
    }
    for (int i = 1; i <= n; i++) {
        for (int j = 1; j <= n; j++) printf("%d ", arr[i][j]);
        printf("\n");
    }
    printf("%d %d", ret.first, ret.second);
}