POJ #2448 A New Operating System

  1 #include <cstdio>
  2 #include <set>
  3 #include <map>
  4 #include <queue>
  5 #define LL long long 
  6 #define MEM second
  7 #define PRO first
  8 #define PID first
  9 #define HP second
 10 using namespace std;
 11 
 12 const int N(1e5+5);
 13 priority_queue<int> que[N];
 14 
 15 typedef pair<int,LL> P;
 16 
 17 map<int,P> mp;
 18 map<int,int> ID;
 19 
 20 bool cmp(const P &a, const P &b){
 21     return a.second!=b.second?a.second>b.second:a.first<b.first;
 22 }
 23 
 24 set<P, bool(*)(const P &, const P &)> o(cmp), m(cmp);
 25 
 26 void Erase(int pid){
 27     int id=ID[pid];
 28     if(que[id].size()){
 29         o.erase(P(pid, (LL)que[id].top()*mp[pid].PRO));
 30         que[id].pop();
 31     }
 32 }
 33 
 34 void remove_process(int pid){
 35     Erase(pid);
 36     m.erase(P(pid, mp[pid].MEM));
 37     int id=ID[pid];
 38     for(; que[id].size(); que[id].pop());
 39     mp.erase(pid);
 40 }
 41 
 42 void Insert(int pid){
 43     int id=ID[pid];
 44     if(que[id].size()){
 45         o.insert(P(pid, (LL)que[id].top()*mp[pid].PRO));
 46     }
 47 }
 48 
 49 void close_process(int pid){
 50     if(mp.find(pid)==mp.end()) puts("Error");
 51     else remove_process(pid);
 52 }
 53 
 54 void close_max_memory(){
 55     if(m.empty()) puts("Empty");
 56     else remove_process(m.begin()->PID);    //error-prone
 57 }
 58 
 59 void run_process(int pid){
 60     if(mp.find(pid)==mp.end()){
 61         puts("Error");
 62         return;
 63     }
 64     int id=ID[pid];
 65     if(que[id].empty()) puts("Empty");
 66     else{
 67         printf("Run Process: %d\n", que[id].top());
 68         Erase(pid), Insert(pid);
 69     }
 70 }
 71 
 72 void modify_memory(int pid, int mem){
 73     if(mp.find(pid)==mp.end()){
 74         puts("Error");
 75         return;
 76     }
 77     if(mp[pid].MEM+mem<=0)
 78         remove_process(pid);
 79     else{
 80         P now=mp[pid];
 81         m.erase(P(pid, now.MEM));
 82         m.insert(P(pid, now.MEM+mem));
 83         mp[pid].MEM+=mem;
 84     }
 85 }
 86 
 87 void change_priority(int pid, int pro){
 88     if(mp.find(pid)==mp.end()){
 89         puts("Error");
 90         return;
 91     }
 92     int id=ID[pid];
 93     if(que[id].size()){
 94         int i_pro=que[id].top();
 95         o.erase(P(pid, (LL)mp[pid].PRO*i_pro));
 96         o.insert(P(pid, (LL)pro*i_pro));
 97     }
 98     mp[pid].PRO=pro;
 99 }
100 
101 void run(){
102     if(o.empty()) puts("Empty");
103     else{
104         printf("Run: %lld\n", o.begin()->HP);
105         int pid=o.begin()->PID;
106         Erase(pid), Insert(pid);
107     }
108 }
109 
110 void add_message(int pid, int pro){
111     if(mp.find(pid)==mp.end()){
112         puts("Error");
113         return;
114     }
115     int id=ID[pid];
116     if(que[id].size()){
117         int i_pro=que[id].top();
118         o.erase(P(pid, (LL)i_pro*mp[pid].PRO));
119     }
120     que[id].push(pro);
121     o.insert(P(pid, (LL)que[id].top()*mp[pid].PRO));
122 }
123 
124 int tail;
125 
126 void create_process(int pid, int mem, int pro){
127     if(mp.find(pid)!=mp.end()){
128         puts("Error");
129         return;
130     }
131     if(mem){
132         mp[pid]=P(pro, mem);
133         m.insert(P(pid, mem));
134         ID[pid]=tail++;
135     }
136 }
137 
138 char s[100];
139 
140 int main(){
141     int n, pid, mem, pro;
142     for(scanf("%d", &n); n--; ){
143         scanf("%s", s);
144         if(s[0]==‘C‘){
145             if(s[1]==‘r‘){
146                 sscanf(s, "CreateProcess(%d,%d,%d)", &pid, &mem, &pro);
147                 create_process(pid, mem, pro);
148             }
149             else if(s[1]==‘l‘){
150                 if(s[5]==‘M‘)
151                     close_max_memory();
152                 else{
153                     sscanf(s, "CloseProcess(%d)", &pid);
154                     close_process(pid);
155                 }
156             }
157             else{
158                 sscanf(s, "ChangePriority(%d,%d)", &pid, &pro);
159                 change_priority(pid, pro);
160             }
161         }
162         else if(s[0]==‘A‘){
163             sscanf(s, "AddMessage(%d,%d)", &pid, &pro);
164             add_message(pid, pro);
165         }
166         else if(s[0]==‘G‘){
167             sscanf(s, "GetMemory(%d,%d)", &pid, &mem);
168             modify_memory(pid, mem);
169         }
170         else if(s[0]==‘F‘){
171             sscanf(s, "FreeMemory(%d,%d)", &pid, &mem);
172             modify_memory(pid, -mem);
173         }
174         else{
175             if(s[3]){
176                 sscanf(s, "RunProcess(%d)", &pid);
177                 run_process(pid);
178             }
179             else run();
180         }
181     }
182     return 0;
183 }