공 넣기 풀이 (Simulation)
문제
https://www.acmicpc.net/problem/10810
문제내용
도현이는 바구니를 총 N개 가지고 있고, 각각의 바구니에는 1번부터 N번까지 번호가 매겨져 있다. 또, 1번부터 N번까지 번호가 적혀있는 공을 매우 많이 가지고 있다. 가장 처음 바구니에는 공이 들어있지 않으며, 바구니에는 공을 1개만 넣을 수 있다.
도현이는 앞으로 M번 공을 넣으려고 한다. 도현이는 한 번 공을 넣을 때, 공을 넣을 바구니 범위를 정하고, 정한 바구니에 모두 같은 번호가 적혀있는 공을 넣는다. 만약, 바구니에 공이 이미 있는 경우에는 들어있는 공을 빼고, 새로 공을 넣는다. 공을 넣을 바구니는 연속되어 있어야 한다.
공을 어떻게 넣을지가 주어졌을 때, M번 공을 넣은 이후에 각 바구니에 어떤 공이 들어 있는지 구하는 프로그램을 작성하시오.
입력
첫째 줄에 N (1 ≤ N ≤ 100)과 M (1 ≤ M ≤ 100)이 주어진다.
둘째 줄부터 M개의 줄에 걸쳐서 공을 넣는 방법이 주어진다. 각 방법은 세 정수 i j k로 이루어져 있으며, i번 바구니부터 j번 바구니까지에 k번 번호가 적혀져 있는 공을 넣는다는 뜻이다. 예를 들어, 2 5 6은 2번 바구니부터 5번 바구니까지에 6번 공을 넣는다는 뜻이다. (1 ≤ i ≤ j ≤ N, 1 ≤ k ≤ N)
도현이는 입력으로 주어진 순서대로 공을 넣는다.
출력
1번 바구니부터 N번 바구니에 들어있는 공의 번호를 공백으로 구분해 출력한다. 공이 들어있지 않은 바구니는 0을 출력한다.
예제 입력 1
5 4
1 2 3
3 4 4
1 4 1
2 2 2
예제 출력 2
1 2 1 1 0
문제풀이
문제해설
- 배열을 응용하면 간단하게 해결할 수 있는 문제이다.
- 바구니를 배열, 바구니 번호를 인덱스 번호, 공의 숫자를 배열값으로 생각하면서 문제를 해결한다.
- 문제에 나오는 조건을 해석하면 i번부터 j번까지 for문을 돌면서 해당 배열에 k값을 넣는 것이다.
코드 설명 1
// 바구니
int v[101] = { 0, };
// 바구니 갯수, 공 넣을 횟수
int N, M;
// 바구니 범위, 넣을 공 번호
int i, j, k;
// 데이터 입력
cin >> N >> M;
- 필요한 변수를 선언한다.
- 배열의 크기가 n이면 0~n-1까지 표현하기 때문에 1부터 n까지 표현하려면 배열의 크기가 n+1이 되어야 한다.
- 문제에서 N의 최대는 100이기 때문에 배열의 크기는 101로 정한다.
- 바구니에 공이 없다는 표시로 전체를 0으로 초기화한다.
코드 설명 2
// 바구니에 공을 넣는 작업
while (M > 0) {
// 공 넣을 범위와 공 번호 입력
cin >> i >> j >> k;
for (int idx = i; idx <= j; idx++) {
v[idx] = k;
}
M--;
}
- 공 넣는 회수 반복은 변수 이름을 적게 생각하기 위해 for문 대신 while문을 선택
- 인덱스 번호가 i~j까지 해당하는 배열에 k값을 넣는다.
// 결과 출력
for (int idx = 1; idx <= N; idx++) {
cout << v[idx]<<" ";
}
- 공의 번호를 공백으로 구분해 출력하는 조건 주의
전체코드
/*
아이디어
- 배열이 바구니 역할
- i~j까지 for문을 돌면서 해당 바구니에 공을 넣는다.
자료구조
- 배열
*/
#include <iostream>
using namespace std;
int main() {
// 바구니
int v[101] = { 0, };
// 바구니 갯수, 공 넣을 횟수
int N, M;
// 바구니 범위, 넣을 공 번호
int i, j, k;
// 데이터 입력
cin >> N >> M;
if (N < 1 || N > 100) return 0;
if (M < 1 || M > 100) return 0;
// 바구니에 공을 넣는 작업
while (M > 0) {
// 공 넣을 범위와 공 번호 입력
cin >> i >> j >> k;
for (int idx = i; idx <= j; idx++) {
v[idx] = k;
}
M--;
}
// 결과 출력
for (int idx = 1; idx <= N; idx++) {
cout << v[idx]<<" ";
}
return 0;
}