[3991] 한번 쏘면 멈출 수 없어

https://www.acmicpc.net/problem/3991

3991번: 한번 쏘면 멈출 수 없어

가장 왼쪽 아래부터 차곡차곡 담아주면 된다.

이전색이 정확히 임의의 열에서(0행에서) 끝나든, 다음 열의 어느 행에서 끝나든 관계없이

색의 순서대로 플레이어가 선택한다면, 항상 최고점 획득이 가능함을 알 수 있다.

---

#include <bits/stdc++.h>

using namespace std;

#ifdef ONLINE_JUDGE

constexpr bool local = false;

#else

constexpr bool local = true;

#endif

using ll = long long;

using pi = pair<ll, ll>;

int n, m, c, a[10][10];

pi getNextPos(pi cur) {

    int ny = cur.first - 1, nx = cur.second;

    if (ny == -1) {

        ny = n - 1;

        nx++;

    }

    return {ny, nx};

}

int main(void) {

    if (!local) ios_base::sync_with_stdio(0), cin.tie(0);

    cin >> n >> m >> c;

    int i, j;

    pi cur = {n, 0};

    for (i = 1; i <= c; i++) {

        int x;

        cin >> x;

        for (j = 0; j < x; j++) {

            cur = getNextPos(cur);

            a[cur.first][cur.second] = i;

        }

    }

    for (i = 0; i < n; i++) {

        for (j = 0; j < m; j++) cout << a[i][j];

        cout << '\n';

    }

    return 0;

}

---