[C++] 백준 19941번: 햄버거 분배(2020정보올림피아드 중등부1번)

문제링크: koi.or.kr/assets/koi/2020/1/problems/m2-problems.pdf 또는 www.acmicpc.net/problem/19941

 

input

식탁의 길이 N 과 햄버거를 선택할 수 있는 거리 k

사람과 햄버거의 위치가 P(사람) H(햄버거) 로 이루어지는 길이 N 의 문자열

 

output

햄버거를 먹을수 있는 최대 사람 수

 

solution 

P 와 H 로 이루어진 문자열을 처음부터 탐색한다.

P 가 나오면 그 지점(i)을 기준으로 i-k 부터 i+k 까지 탐색한다.

탐색중에 햄버거가 발견되었다면 햄버거를 H나 P가 아닌 다른 문자로 바꾸고 햄버거를 먹을 수 있는 사람 수를 1 더해준다.

 

code

	#include <iostream>
	#include <vector>
	using namespace std;
	int main()
	{
	int N, k, cnt = 0;
	char c;
	vector<char> ph;
	cin >> N >> k;
	for(int i = 0; i < N; i++){
	cin >> c;
	ph.push_back(c);
	}
	for(int i = 0; i < N; i++){
	if(ph[i] != 'P')continue;
	for(int j = i - k; j <= i + k; j++){
	if(0 <= j && j < N && ph[j] == 'H'){
	ph[j] = '0';
	cnt++;
	break;
	}
	}
	}
	cout << cnt;
	return 0;
	}

view raw hamburger.cpp hosted with ❤ by GitHub