#include <iostream> #include <algorithm> #include <vector> using namespace std; int main() { ios_base::sync_with_stdio(0); int n,m,l=0; cin>>n>>m; int bag[n],plec[m]; vector <int> v[m]; for(int i=0;i<n;i++) cin>>bag[i]; for(int i=0;i<m;i++) cin>>plec[i]; sort(bag, bag+n); sort(plec, plec+m); for(int i=n-1;i>=0;i--) { for(int j=0;j<m;j++) { if(plec[j]>=bag[i]) { v[j].push_back(bag[i]); plec[j]-=bag[i]; j=m; } } } /*for(int i=0;i<n;i++) cout<<bag[i]<<" "; cout<<endl; for(int i=0;i<m;i++) cout<<plec[i]<<" "; cout<<endl;*/ /*for(int i=0;i<m;i++) { for(int j=0;j<v[i].size();j++) { cout<<v[i][j]<<" "; } cout<<" "<<v[i].size(); cout<<endl; }*/ for(int i=m-1;i>0;i--) { for(int j=i-1;j>=0;j--) { while(v[j].size()>0 && plec[i]>=v[j].back()) { //cout<<" ^^^^^^^^^^^"<<endl; //cout<<v[j].back()<<" "<<plec[i]<<endl; plec[i]-=v[j].back(); plec[j]+=v[j].back(); v[i].push_back(v[j].back()); v[j].pop_back(); //cout<<plec[j]<<" "<<plec[i]<<endl; } } } for(int i=0;i<m;i++) { l+=v[i].size(); } if(l!=n) cout<<"NIE"; else { int wynik=0; for(int i=0;i<m;i++) if(v[i].size()>0) wynik++; cout<<wynik; } //for(int i=0;i<m;i++) cout<<plec[i]<<" "; /*for(int i=0;i<m;i++) { for(int j=0;j<v[i].size();j++) { cout<<v[i][j]<<" "; } cout<<v[i].back(); cout<<endl; }*/ 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 | #include <iostream> #include <algorithm> #include <vector> using namespace std; int main() { ios_base::sync_with_stdio(0); int n,m,l=0; cin>>n>>m; int bag[n],plec[m]; vector <int> v[m]; for(int i=0;i<n;i++) cin>>bag[i]; for(int i=0;i<m;i++) cin>>plec[i]; sort(bag, bag+n); sort(plec, plec+m); for(int i=n-1;i>=0;i--) { for(int j=0;j<m;j++) { if(plec[j]>=bag[i]) { v[j].push_back(bag[i]); plec[j]-=bag[i]; j=m; } } } /*for(int i=0;i<n;i++) cout<<bag[i]<<" "; cout<<endl; for(int i=0;i<m;i++) cout<<plec[i]<<" "; cout<<endl;*/ /*for(int i=0;i<m;i++) { for(int j=0;j<v[i].size();j++) { cout<<v[i][j]<<" "; } cout<<" "<<v[i].size(); cout<<endl; }*/ for(int i=m-1;i>0;i--) { for(int j=i-1;j>=0;j--) { while(v[j].size()>0 && plec[i]>=v[j].back()) { //cout<<" ^^^^^^^^^^^"<<endl; //cout<<v[j].back()<<" "<<plec[i]<<endl; plec[i]-=v[j].back(); plec[j]+=v[j].back(); v[i].push_back(v[j].back()); v[j].pop_back(); //cout<<plec[j]<<" "<<plec[i]<<endl; } } } for(int i=0;i<m;i++) { l+=v[i].size(); } if(l!=n) cout<<"NIE"; else { int wynik=0; for(int i=0;i<m;i++) if(v[i].size()>0) wynik++; cout<<wynik; } //for(int i=0;i<m;i++) cout<<plec[i]<<" "; /*for(int i=0;i<m;i++) { for(int j=0;j<v[i].size();j++) { cout<<v[i][j]<<" "; } cout<<v[i].back(); cout<<endl; }*/ return 0; } |