[백준] 17612 - 쇼핑몰

17612 백준 쇼핑몰

쇼핑몰

문제

대형 쇼핑몰에서 쇼핑을 마친 N명의 고객들이 계산을 하고 쇼핑몰을 떠나고자 계산대 앞에 줄을 서 있다. 각 고객은 커다란 짐수레(cart)에 물건을 담아 계 산대로 간다. 쇼핑몰에는 아래 그림과 같이 K개의 계산대가 병렬로 배치되어 있다.  쇼핑몰 안내원들은 계산대에 온 사람들을 가장 빨리 계산을 마칠 수 있는 계산대로 안내를 한다. 안내원은 각 계산대에서 기다리고 있는 사람들이 계산을 하는데 얼마나 걸리는지 이미 알고 있다.

안내원이 고객을 계산대로 안내할 때 두 계산대에서 기다려야 할 시간이 같을 경우에는 가장 번호가 작은 계산대로 안내를 한다. 즉 3번, 5번 계산대에서 기다릴 시간이 똑같이 15분으로 최소일 경우에는 3번으로 안내를 한다.

계산을 마친 고객은 출구를 통하여 쇼핑몰을 완전히 빠져 나간다. 만일 계산대에서 계산을 마친 고객의 시간이 동일하다면 출구에 가까운 높은 번호 계산대의 고객부터 먼저 빠져나간다. 예를 들어 두 계산대 4번과 10번에서 두 고객이 동시에 계산을 마쳤다면 계산대의 번호가 더 높은 10번 계산대의 고객이 먼저 쇼핑몰을 나간다. 물건을 계산하는 데에는 종류에 관계없이 동일하게 1분이 소요된다. 즉, 물건이 w개 있는 손님이 계산을 마치는 데에는 정확히 w분이 소요된다.

여러분은 계산대로 들어가기 위하여 줄을 서 있는 고객 N명의 정보( 회원번호, 구매한 물건의 수)를 알고 있을 때, 이들이 계산을 하고 쇼핑몰을 빠져나오는 순서를 구해야 한다. 계산대로 들어가고 계산대에서 나오는데 걸리는 시간은 없다고 가정한다.

입력

입력의 첫 줄에는 2개의 정수 N(1 ≤ N ≤ 100,000)과 k(1 ≤ k ≤ 100,000)가 주어진다. 다음 줄부터 N개의 줄에 걸쳐 고객 N명의 정보가 줄 맨 앞의 고객부터 맨 뒤 고객까지 순서대로 주어진다. i번째 줄은 줄의 앞에서 i번째 고객의 회원번호 idi(1 ≤ idi ≤ 1,000,000)와 그가 구입한 물건의 수 wi(1 ≤ wi ≤ 20)로 이루어져 있다. N명의 회원번호는 모두 다르다.

출력

고객 N명의 회원번호를 쇼핑몰을 빠져나가는 순서대로 r1, r2, ..., rN이라 할 때, 1×r1 + 2×r2 + ... + N×rN의 값을 출력한다. 출력값이 int 범위를 넘어갈 수 있음에 유의하라.

 

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

int N, k;

struct Person {
	int id, w, totalMin, casher;
	bool operator< (const Person& otherPerson) const {
		if (totalMin < otherPerson.totalMin)
			return true;
		else if (totalMin == otherPerson.totalMin) {
			if (casher < otherPerson.casher)
				return false;
			else
				return true;
		}
		else
			return false;
	}
};

struct Casher {
	int id, totalMin;
	bool operator< (const Casher& other) const {
		if (totalMin < other.totalMin)
			return false;
		else if (totalMin == other.totalMin) {
			if (id < other.id)
				return false;
			else 
				return true;
		}
		else
			return true;
	}
};

int main() {
	ios_base::sync_with_stdio(false);
	cin.tie(nullptr);
	cout.tie(nullptr);

	cin >> N >> k;

	priority_queue<Casher> casherPQ;
	for (int i = 1; i <= k; i++) {
		casherPQ.push({ i, 0 });
	}

	vector<Person> customerVec(N);
	int cusId, weight;
	for (int i = 0; i < N; i++) {
		cin >> cusId >> weight;
		customerVec[i] = { cusId, weight, 0, 0 };
	}

	for (int i = 0; i < N; i++) {
		Casher topPriCasher = casherPQ.top(); casherPQ.pop();
		int tmpId = topPriCasher.id;
		int tmpTotalMin = topPriCasher.totalMin;

		tmpTotalMin += customerVec[i].w;
		customerVec[i].totalMin = tmpTotalMin;
		customerVec[i].casher = tmpId;

		casherPQ.push({ tmpId, tmpTotalMin });
	}
	sort(customerVec.begin(), customerVec.end());

	long long ret = 0;
	for (int i = 1; i <= N; i++) {
		ret += i * (long long)customerVec[i - 1].id;
	}

	cout << ret;

	return 0;
}

고객과 계산대를 연결시키되, 고객이 계산대로 가기보다는, 계산대가 고객에게 간다고 생각하면 문제 풀기가 수월해진다.