과제 진행하기 C++

https://school.programmers.co.kr/learn/courses/30/lessons/176962

프로그래머스

 

※ 문제

과제를 받은 루는 다음과 같은 순서대로 과제를 하려고 계획을 세웠습니다.
과제는 시작하기로 한 시각이 되면 시작합니다.
새로운 과제를 시작할 시각이 되었을 때, 기존에 진행 중이던 과제가 있다면 진행 중이던 과제를 멈추고 새로운 과제를 시작합니다.
진행중이던 과제를 끝냈을 때, 잠시 멈춘 과제가 있다면, 멈춰둔 과제를 이어서 진행합니다.
만약, 과제를 끝낸 시각에 새로 시작해야 되는 과제와 잠시 멈춰둔 과제가 모두 있다면, 새로 시작해야 하는 과제부터 진행합니다.
멈춰둔 과제가 여러 개일 경우, 가장 최근에 멈춘 과제부터 시작합니다.
과제 계획을 담은 이차원 문자열 배열 plans가 매개변수로 주어질 때, 과제를 끝낸 순서대로 이름을 배열에 담아 return 하는 solution 함수를 완성해주세요.

 

※ 나의 풀이

이 문제는 평범한 구현 문제 같았다.

다만, 구현하기 굉장히 귀찮은 문제이다.

부등호 하나를 잘못적용해서 시간을 엄청 잡아먹었다...

 

 

struct Edge
{
	string subject;
	int start_time = 0, need_time = 0;

	Edge(string s, string t, string n)
	{
		subject = s;
		start_time = s2i(t);
		need_time = stoi(n);
	}

	Edge(string s, int t, int n)
	{
		subject = s;
		start_time = t;
		need_time = n;
	}

	int s2i(string time)
	{
		int hour = stoi(time.substr(0, 2));
		int minute = stoi(time.substr(3, 5));
		return hour * 60 + minute;
	}

	bool operator<(const Edge a) const
	{
		return start_time < a.start_time;
	}
};

	vector<string> answer;

	// 과목 정렬
	vector<Edge> info;
	for (int i = 0; i < plans.size(); i++)
	{
		info.push_back(Edge(plans[i][0], plans[i][1], plans[i][2]));
	}
	sort(info.begin(), info.end());

	// 잔여 과목 저장소
	stack<Edge> st;

	// 과목 처리하기
	int extra_time = 0;
	for (int i = 0; i < info.size() - 1; i++)
	{
		Edge cur = info[i];
		Edge next = info[i + 1];
		extra_time = next.start_time - cur.start_time;

		if (extra_time == 0)
		{
			answer.push_back(cur.subject);
			continue;
		}

		if (extra_time < cur.need_time)
		{
			st.push(Edge(cur.subject, cur.start_time, cur.need_time - extra_time));
		}
		else // extra_time >= cur.need_time
		{
			answer.push_back(cur.subject);
			extra_time -= cur.need_time;
			while (!st.empty() && extra_time != 0)
			{
				if (extra_time >= st.top().need_time)
				{
					extra_time -= st.top().need_time;
					answer.push_back(st.top().subject);
					st.pop();
				}
				else // extra_time < st.top().need_time
				{
					st.top().need_time -= extra_time;
					break;
				}
			}
		}
	}
	answer.push_back(info[info.size() - 1].subject);

	while (!st.empty())
	{
		answer.push_back(st.top().subject);
		st.pop();
	}