Advertisement
Korotkodul

N4 2

Nov 4th, 2022 (edited)
656
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 3.31 KB | None | 0 0
  1. #include <iostream>
  2. #include <cmath>
  3. #include <vector>
  4. #include <queue>
  5. #include <algorithm>
  6. #include <string>
  7. #include <stack>
  8. #include <set>
  9. #include <map>
  10. #define pii pair <int, int>
  11. #define pb(x) push_back(x)
  12. using namespace std;
  13. using ll = long long;
  14. using ld = long double;
  15. using db = double;
  16. void cv(vector <int> &v) {
  17.     for (auto x : v) cout << x << ' ';
  18.     cout << "\n";
  19. }
  20.  
  21. void cvl(vector <ll> &v) {
  22.     for (auto x : v) cout << x << ' ';
  23.     cout << "\n";
  24. }
  25.  
  26.  
  27. void cvv(vector <vector <int> > &v) {
  28.     for (auto x : v) cv(x);
  29.     cout << "\n";
  30. }
  31.  
  32. void cvb(vector <bool> v) {
  33.     for (bool x : v) cout << x << ' ';
  34.     cout << "\n";
  35. }
  36.  
  37. void cvs(vector <string>  v) {
  38.     for (auto a : v) {
  39.         cout << a << "\n";
  40.     }
  41. }
  42.  
  43. void cvp(vector <pii> a) {
  44.     for (auto p : a) {
  45.         cout << p.first << ' ' << p.second << "\n";
  46.     }
  47.     cout << "\n";
  48. }
  49. #include <ctime>
  50.  
  51. bool sh = 1;
  52. int N, n;
  53.  
  54. vector <int> gen() {
  55.     srand(time(0));
  56.     n = rand() % 5 + 2;
  57.     vector <int> a;
  58.     for (int i = 0; i < n; ++i) {
  59.         int x = rand() % 20 + 1;
  60.         a.pb(x);
  61.         N += x;
  62.     }
  63.     sort(a.begin(), a.end());
  64.     reverse(a.begin(), a.end());
  65.     //cv(a);
  66.     return a;
  67. }
  68.  
  69. bool ok(vector <int> a) {
  70.     for (int i = 0; i < a.size() - 1; ++i) {
  71.         if (a[i] == a[i + 1]) {
  72.             return 0;
  73.         }
  74.     }
  75.     return 1;
  76. }
  77.  
  78. vector <int> my(vector <int> a) {
  79.     int dif = a[0] - a[1] - 1;
  80.     a[0] -= dif;
  81.     vector <int> ans(N);
  82.     int id = -1, cnt = 0;
  83.     int to = n;
  84.     if (sh) {
  85.         cout << "A\n";
  86.     }
  87.     while (cnt < N) {
  88.         int j = -1;
  89.         if (sh) {
  90.             cout << "B\n";
  91.         }
  92.         while (j + 1 < to) {
  93.             id++;
  94.             if (sh) cout << "id j = " << id << ' ' << j << "\n";
  95.             if (sh) {
  96.                 cout << "a\n";
  97.                 cv(a);
  98.                 cout << "dif = " << dif << "\n";
  99.             }
  100.             if (j == -1) {
  101.                 ans[id] = 0;
  102.                 j = 0;
  103.                 a[0]--;
  104.                 cnt++;
  105.                 continue;
  106.             }
  107.             if (dif > 0) {
  108.                 if (sh) {
  109.                     cout << "del dif\n";
  110.                 }
  111.                 if (id == 0 || id > 0 && ans[id - 1] != 0) {
  112.                     ans[id] = 0;
  113.                     dif--;
  114.                     cnt++;
  115.                     continue;
  116.                 }
  117.             }
  118.             j++;
  119.             a[j]--;
  120.             cnt++;
  121.             if (sh) {
  122.                 cout << "just go\n";
  123.             }
  124.             ans[id] = j;
  125.             if (a[j] == 0) {
  126.                 to = min(to, j);
  127.             }
  128.         }
  129.     }
  130.     if (ok(ans)) {
  131.         return ans;
  132.     }
  133.     return {-1};
  134. }
  135.  
  136. bool random = 1;
  137.  
  138. int main() {
  139.     /*ios::sync_with_stdio(0);
  140.     cin.tie(0);
  141.     cout.tie(0);*/
  142.  
  143.     vector <int> a;
  144.     if (sh && random) {
  145.         a = gen();
  146.     } else if (sh) {
  147.         cin >> n;
  148.         a.resize(n);
  149.         for (int &i: a) {
  150.                 cin >> i;
  151.                 N += i;
  152.         }
  153.     }
  154.  
  155.     if (sh) {
  156.         cout << "N = " << N <<"\n";
  157.         cout << "a\n";
  158.         cv(a);
  159.     }
  160.  
  161.     vector <int> r = my(a);
  162.     if (sh) {
  163.         cout << "r\n";
  164.         cv(r);
  165.     }
  166. }
  167. /*
  168. N = 58
  169. a
  170.  
  171. 5
  172. 18 17 11 8 4
  173. */
  174.  
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement