<8-4> 그리디 - 배치 시스템 (BatchSystem)
PS/Topcoder training2018. 7. 16. 01:54
### Greedy ###
### SRM 481 Div2 Level 3 ###
<소스코드>
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 | #include <algorithm> #include <string> #include <vector> #include <map> using namespace std; typedef long long int lli; struct P { int i, duration; P(int i, int duration) : i(i), duration(duration) {} }; struct userInfo { vector<P> v; lli sum; userInfo(vector<P> v, lli sum) : v(v), sum(sum) {} }; bool cmp(P a, P b) { return a.i < b.i; } bool cmp2(userInfo a, userInfo b) { return a.sum != b.sum ? a.sum < b.sum : a.v[0].i < b.v[0].i; } class BatchSystem { public: vector<int> schedule(vector <int> duration, vector <string> user) { int userIdx=0; map<string, int> msi; vector<userInfo> info; for(int i=0; i<duration.size(); i++) { if(msi.find(user[i]) == msi.end()) { msi[user[i]] = ++userIdx; vector<P> v; v.push_back(P(i, duration[i])); info.push_back(userInfo(v, duration[i])); } else { int idx = msi[user[i]]-1; info[idx].sum += duration[i]; info[idx].v.push_back(P(i, duration[i])); } } for(int i=0; i<info.size(); i++) { sort(info[i].v.begin(), info[i].v.end(), cmp); } sort(info.begin(), info.end(), cmp2); vector<int> ans; for(int i=0; i<info.size(); i++) { for(int j=0; j<info[i].v.size(); j++) { ans.push_back(info[i].v[j].i); } } return ans; } }; | cs |
'PS > Topcoder training' 카테고리의 다른 글
<5-8> 전체탐색 - 미로 만드는 사람 (MazeMaker) (0) | 2018.07.16 |
---|---|
<8-8> 수학 - 둥근 모양의 국가들 (CirclesCountry) (0) | 2018.07.16 |
<8-1> 탐색범위 한정 - 다양한 색상의 상자와 공 (ColorfulBoxesAndBalls) (0) | 2018.07.14 |
<7-5> DP - 악수 (HandShaking) (0) | 2018.07.04 |
<5-7> 전체탐색 - 고장난 로봇 (CrazyBot) (0) | 2018.07.03 |