#include<iostream> #include<vector> using namespace std; struct Ryba { int waga; Ryba(int w) { waga = w; } }; void BubbleSortVector(vector<Ryba> &ryba) { for(int i=0;i<ryba.size();i++) { for(int j=1;j<ryba.size();j++) { if(ryba[j].waga<ryba[j-1].waga) { swap(ryba[j].waga,ryba[j-1].waga); } } } } int FindInVector(vector<Ryba> ryba,int w) { for(int i=0;i<ryba.size();i++) { if(ryba[i].waga == w) return i; } return -1; } int WpierdalanieRyby(vector<Ryba> ryba,int w,int d) { int ans = 0; bool zjadl = false; for(int i=ryba.size()-1;i>=0;i--) { if(w>=d) { zjadl = true; break; } //cout<<"waga szczupola: "<<w<<endl; if(ryba[i].waga!=-1 && ryba[i].waga<w) { //cout<<"wcinam rybe o wadze: "<<ryba[i].waga<<endl; w+=ryba[i].waga; ryba[i].waga=-1; ans++; i=ryba.size(); } } return (zjadl ? ans : -1); } int main() { int n; cin>>n; vector<Ryba> ryba; for(int i=0;i<n;i++) { int w; cin>>w; ryba.push_back(Ryba(w)); } BubbleSortVector(ryba); int q; cin>>q; for(int i=0;i<q;i++) { /*cout<<"aktualne ryby: "; for(int i=0;i<ryba.size();i++) { cout<<ryba[i].waga<<" "; } cout<<endl;*/ int l; cin>>l; if(l==1) { //cout<<"atak szczupola"<<endl; int s,k; cin>>s>>k; int ile = WpierdalanieRyby(ryba,s,k); cout/*<<"ODP: "*/<<ile<<endl; } else if(l==2) { //cout<<"ryba in"<<endl; int w; cin>>w; ryba.push_back(Ryba(w)); BubbleSortVector(ryba); } else if(l==3) { //cout<<"ryba out"<<endl; int w; cin>>w; int index = FindInVector(ryba,w); ryba[index].waga = -1; BubbleSortVector(ryba); } } return 0; }
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 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 | #include<iostream> #include<vector> using namespace std; struct Ryba { int waga; Ryba(int w) { waga = w; } }; void BubbleSortVector(vector<Ryba> &ryba) { for(int i=0;i<ryba.size();i++) { for(int j=1;j<ryba.size();j++) { if(ryba[j].waga<ryba[j-1].waga) { swap(ryba[j].waga,ryba[j-1].waga); } } } } int FindInVector(vector<Ryba> ryba,int w) { for(int i=0;i<ryba.size();i++) { if(ryba[i].waga == w) return i; } return -1; } int WpierdalanieRyby(vector<Ryba> ryba,int w,int d) { int ans = 0; bool zjadl = false; for(int i=ryba.size()-1;i>=0;i--) { if(w>=d) { zjadl = true; break; } //cout<<"waga szczupola: "<<w<<endl; if(ryba[i].waga!=-1 && ryba[i].waga<w) { //cout<<"wcinam rybe o wadze: "<<ryba[i].waga<<endl; w+=ryba[i].waga; ryba[i].waga=-1; ans++; i=ryba.size(); } } return (zjadl ? ans : -1); } int main() { int n; cin>>n; vector<Ryba> ryba; for(int i=0;i<n;i++) { int w; cin>>w; ryba.push_back(Ryba(w)); } BubbleSortVector(ryba); int q; cin>>q; for(int i=0;i<q;i++) { /*cout<<"aktualne ryby: "; for(int i=0;i<ryba.size();i++) { cout<<ryba[i].waga<<" "; } cout<<endl;*/ int l; cin>>l; if(l==1) { //cout<<"atak szczupola"<<endl; int s,k; cin>>s>>k; int ile = WpierdalanieRyby(ryba,s,k); cout/*<<"ODP: "*/<<ile<<endl; } else if(l==2) { //cout<<"ryba in"<<endl; int w; cin>>w; ryba.push_back(Ryba(w)); BubbleSortVector(ryba); } else if(l==3) { //cout<<"ryba out"<<endl; int w; cin>>w; int index = FindInVector(ryba,w); ryba[index].waga = -1; BubbleSortVector(ryba); } } return 0; } |