[C++/groom] Lv.2 : 장마

 

https://level.goorm.io/exam/194982/%EC%9E%A5%EB%A7%88/quiz/1

구름LEVEL

---

구름 레벨 2 구현문제.

 

범위가 n은 1000, m은 100000까지라서 아슬아슬하게 이중반복문 사용이 가능하다 ㅋㅋ

단순 구현이지만 고려해야할 사항이 많아서 까다로웠다!

 

 

나의 풀이

#include <iostream>
#include <vector>
using namespace std;
int main() {
	int n, m, tmp1, tmp2;
	vector<int> ori_list;
	vector<int> k_list;
	vector<int> rain_in_2days;
	vector<pair<int,int>> rain_place;
	// 입력 로직
	cin >> n >> m;
	for(int i=0; i<n; ++i) {
		cin >> tmp1;
		ori_list.push_back(tmp1);
		k_list.push_back(tmp1);
		rain_in_2days.push_back(0);
	}
	for(int i=0; i<m; ++i) {
		cin >> tmp1 >> tmp2;
		rain_place.push_back({tmp1, tmp2});
	}
	
	int day = 0;
	while(day < m){
		day++; // 1일차부터 시작
		int s = rain_place[day-1].first - 1;
		int e = rain_place[day-1].second - 1;
		for(int i=0; i<n; ++i) {
			if(rain_in_2days[i]>0) rain_in_2days[i]--; // 비온지 하루 지나면 -1
		}
	
		for(int i=s; i<=e; ++i){
			k_list[i]++;
			rain_in_2days[i] = 3; // 비 1일째=3, 2일째=2, 3일째=1
		}
		if(day%3==0){
			for(int i=0; i<n; ++i) {
				if(rain_in_2days[i]>0){
					if(k_list[i] > ori_list[i]) k_list[i]--;
				}
			}
		}
	}

	for(int i=0; i<n; ++i) {
		cout << k_list[i] << " ";
	}
	
	return 0;
}