Koala - 11기/코딩테스트 준비 스터디

[백준/C++] 19941번: 햄버거 분배

나는 푸딩 2023. 9. 3. 15:43

Problem

Solution

1. 먼저 두 개의 정수 N과 K를 입력 받는다.
2. N 길이의 문자열을 입력받는다.
3. 각각의 'P' (사람) 위치에 대해 가장 가까운 햄버거('H')를 찾아서 먹을 수 있는지 확인한다.
4. for 루프를 통해 문자열을 순회하면서 'P' 위치를 찾고, 그 주변에 있는 햄버거 중에서 최대 K 거리 이내에 있는 햄버거를 찾아서 먹는다.
5. 햄버거를 먹었다면, 그 햄버거는 'E' (먹었다는 표시)로 표시되고, 더 이상 해당 햄버거를 먹을 수 없다.
6. 위의 과정을 모든 'P' 위치에 대해 반복하고, 햄버거를 먹은 사람의 수를 계산한다.
7. 최종적으로, 먹을 수 있는 햄버거를 먹은 사람의 수를 출력한다.

Answer

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;

int main() {
    int N, K;
    cin >> N >> K;
    string table;
    cin >> table;

    int maxPeople = 0;

    for (int i = 0; i < N; i++) {
        if (table[i] == 'P') {
            for (int j = max(0, i - K); j <= min(N - 1, i + K); j++) {
                if (table[j] == 'H') {
                    maxPeople++;
                    table[j] = 'E';
                    break;
                }
            }
        }
    }

    cout << maxPeople << endl;
    return 0;
}

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