BOJ)14195 DotB

문제: icpc.me/14195

각 테스트 케이스마다 n과 c가 주어진다. 이후 n개의 시퀀스가 주어진다.

이제 0번 인덱스부터 증가하는 순서로 보면서 시퀀스를 c만큼 감소시킬 것이다. 해당 원소가 0이하가 되면 시퀀스의 탐색 방향을 반대방향으로 바꾸고 해당 원소를 제거한다.

이렇게 하였을 때, n+5번 탐색을 하였을 때 가장 마지막에 감소 된 원소의 번호를 출력하는 문제이다.

n이 적은편이기 때문에 시뮬레이션을 돌려주면 된다.

#include <cstdio>
#include <algorithm>
#include <vector>
using namespace std;
int t, r, n, c;
int func(vector<pair<int, int>> vt, int idx, int dir, int cnt) {
    if (vt.size() == 1)
        return vt[0].first;
    if (cnt == 1)
        return vt[idx].first;
    vt[idx].second -= c;
    if (vt[idx].second <= 0) {
        vt.erase(vt.begin() + idx);
        if (dir) {
            dir = 0;
            if (idx == vt.size())
                idx = 0;
        }
        else {
            idx--;
            dir = 1;
            if (idx == -1)
                idx = vt.size() - 1;
        }
    }
    if (dir)
        return func(vt, (idx + 1) % vt.size(), dir, cnt - 1);
    else
        return func(vt, (idx - 1 + vt.size()) % vt.size(), dir, cnt - 1);
}
int main() {
    scanf("%d", &t);
    while (t--) {
        scanf("%d%d", &n, &c);
        vector<pair<int, int>> vt;
        for (int i = 1; i <= n; i++) {
            int x;
            scanf("%d", &x);
            vt.push_back({ i,x });
        }
        printf("%d\n", func(vt, 0, 1, n + 5));
    }
    return 0;
}