Cod sursa(job #414626)

Utilizator vladiiIonescu Vlad vladii Data 10 martie 2010 12:23:29
Problema Zeap Scor 80
Compilator cpp Status done
Runda Arhiva de probleme Marime 4.43 kb
#include <iostream>
#include <set>
using namespace std;

set<int> S;
set<int>::iterator it, it1, it2, it5;
pair<set<int>::iterator,bool> ret;
multiset<int> D;
multiset<int>::iterator it3, it4;
char op[10];

int main() {
    FILE *f1=fopen("zeap.in", "r"), *f2=fopen("zeap.out", "w");
    int i, j, p, q;
    while(fscanf(f1, "%s", op) != EOF) {
         if(op[0]=='I') {
              fscanf(f1, "%d", &p);
              ret=S.insert(p);
              if(ret.second==true) {
                   it=ret.first;
                   if(S.size()>1) {
                        it5=S.end(); it5--;
                        if(it==S.begin()) {
                             it1=it; it1++;
                             if(D.size()) {
                                  it4=D.find(abs((*it1)-(*it))); 
                                  if(it4!=D.end()) { D.erase(D.find(abs((*it1)-(*it)))); }
                             }
                             q=abs((*it1)-p);
                             D.insert(q);
                        }
                        else if(it==it5) {
                             it1=it; it1--;
                             if(D.size()) { 
                                  it4=D.find(abs((*it1)-p));
                                  if(it4!=D.end()) { D.erase(D.find(abs((*it1)-p))); }
                             }
                             q=abs((*it1)-p);
                             D.insert(q);
                        }
                        else {
                             it1=it; it1--;
                             it2=it; it2++;
                             if(D.size()) { 
                                  it4=D.find(abs((*it2)-(*it1)));        
                                  if(it4!=D.end()) { D.erase(D.find(abs((*it2)-(*it1)))); }
                             }
                             q=abs((*it1)-p);
                             j=abs((*it2)-p);
                             D.insert(q); D.insert(j);
                        }
                   }
                   else if(S.size()==1) { /**do nothing;**/ }
              }
         }
         else if(op[0]=='S') {
              fscanf(f1, "%d", &p);
              it=S.find(p);
              if(it==S.end()) {
                   fprintf(f2, "%d\n", -1);
              }
              else {
                   it5=S.end(); it5--;
                   if(S.size()==1) { S.erase(it); }
                   else if(it==S.begin()) {
                        it1=it; it1++;
                        if(D.size()) {
                             it4=D.find(abs(*it-*it1)); 
                             if(it4!=D.end()) { D.erase(abs(*it-*it1)); }
                        }
                        S.erase(it);
                   }
                   else if(it==it5) {
                        it1=it5; it1--;
                        if(D.size()) { 
                             it4=D.find(abs(*it-*it1)); 
                             if(it4!=D.end()) { D.erase(abs(*it-*it1)); }
                        }
                        S.erase(it);
                   }
                   else {
                        it1=it; it1--;
                        it2=it; it2++;
                        D.insert(abs((*it2)-(*it1)));
                        it3=D.find(abs((*it)-(*it1)));
                        if(it3!=D.end()) { D.erase(it3); }
                        it4=D.find(abs((*it)-(*it2)));
                        if(it4!=D.end()) { D.erase(it4); }
                        S.erase(it);
                   }
              }
         }
         else if(op[0]=='C') {
              fscanf(f1, "%d", &p);
              it=S.find(p);
              if(it==S.end()) {
                   fprintf(f2, "%d\n", 0);
              }
              else {
                   fprintf(f2, "%d\n", 1);
              }
         }
         else if(op[0]=='M' && op[1]=='A') {
              if(S.size()<2) {
                   fprintf(f2, "%d\n", -1);
              }
              else {
                   it1=S.end(); it1--;
                   it2=S.begin();
                   fprintf(f2, "%d\n", (*it1)-(*it2));
              }
         }
         else if(op[0]=='M' && op[1]=='I') {
              if(S.size()<2) {
                   fprintf(f2, "%d\n", -1);
              }
              else {
                   fprintf(f2, "%d\n", (*D.begin()));
              }
         }
    }
    fclose(f1); fclose(f2);
    return 0;
}