Advertisement
pb_jiang

CF1490G WA

Jan 11th, 2023
102
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 3.80 KB | None | 0 0
  1. #include <assert.h>
  2. #include <bits/stdc++.h>
  3. using namespace std;
  4. #define dbg(...) logger(#__VA_ARGS__, __VA_ARGS__)
  5. template <typename... Args> void logger(string vars, Args &&... values)
  6. {
  7.     cerr << vars << " = ";
  8.     string delim = "";
  9.     (..., (cerr << delim << values, delim = ", "));
  10.     cerr << endl;
  11. }
  12.  
  13. template <class T> inline auto vv(int m) { return vector<vector<T>>(m, vector<T>(m)); }
  14. template <class T> inline auto vv(int m, int n) { return vector<vector<T>>(m, vector<T>(n)); }
  15. template <class T, T init> inline auto vv(int m) { return vector<vector<T>>(m, vector<T>(m, init)); }
  16. template <class T, T init> inline auto vv(int m, int n) { return vector<vector<T>>(m, vector<T>(n, init)); }
  17.  
  18. template <class T> using mpq = priority_queue<T, vector<T>, greater<T>>;
  19.  
  20. using ll = long long;
  21. using pii = pair<int, int>;
  22. using vl = vector<ll>;
  23. using vi = vector<int>;
  24.  
  25. int main(int argc, char **argv)
  26. {
  27.     int t;
  28.     cin >> t;
  29.     while (t--) {
  30.         int n, m;
  31.         cin >> n >> m;
  32.         vector<ll> a(n);
  33.         ll acc = 0;
  34.         vector<ll> pref;
  35.         map<ll, int> cmap;
  36.  
  37.         for (int i = 0; i < n; ++i) {
  38.             cin >> a[i];
  39.             acc += a[i];
  40.             if (acc >= 0 && (pref.size() == 0 || acc > pref.back())) {
  41.                 pref.push_back(acc);
  42.                 cmap[acc] = i;
  43.             }
  44.         }
  45.  
  46.         ll x;
  47.         auto calc = [&](ll idx, ll x) { return (x - pref[idx]) / acc + idx; };
  48.         auto check = [&](ll idx, ll x) {
  49.             if (idx == 0)
  50.                 return calc(idx, x) < calc(idx + 1, x);
  51.             if (idx == n - 1)
  52.                 return calc(idx - 1, x) < calc(idx, x);
  53.             return calc(idx - 1, x) < calc(idx, x) && calc(idx, x) < calc(idx + 1, x);
  54.         };
  55.         auto find = [&](ll x) {
  56.             /*
  57.             ll lb = 0, ub = pref.size(), cnt = ub - lb, it, step;
  58.             while (cnt) {
  59.                 it = lb, step = cnt / 2, it += step;
  60.                 if (!check(it, x)) {
  61.                     lb = it + 1;
  62.                     cnt -= step + 1;
  63.                 } else
  64.                     cnt = step;
  65.             }
  66.             return calc(lb, x);
  67.             */
  68.             int l = 0, r = pref.size(), mid = 0;
  69.             while (l < r) {
  70.                 int mid = (l + r) / 2;
  71.                 if (mid == 0) {
  72.                     if (calc(mid, x) <= calc(mid + 1, x)) {
  73.                         l = mid;
  74.                     } else {
  75.                         r = mid + 1;
  76.                     }
  77.                     break;
  78.                 }
  79.                 if (mid == n - 1) {
  80.                     if (calc(mid, x) <= calc(mid - 1, x)) {
  81.                         l = mid;
  82.                     } else {
  83.                         r = mid - 1;
  84.                     }
  85.                     break;
  86.                 }
  87.                 ll mm1 = calc(mid - 1, x), mm = calc(mid, x), mp1 = calc(mid, x);
  88.                 if (mm1 >= mm && mm <= mp1) {
  89.                     l = mid;
  90.                     break;
  91.                 }
  92.                 if (mm1 < mm && mm < mp1) {
  93.                     r = mid - 1;
  94.                 } else if (mm1 > mm && mm > mp1) {
  95.                     l = mid + 1;
  96.                 }
  97.             }
  98.             return calc(l, x);
  99.         };
  100.         for (int i = 0; i < m; ++i) {
  101.             cin >> x;
  102.             int lb = lower_bound(pref.begin(), pref.end(), x) - pref.begin();
  103.             if (lb != pref.size()) {
  104.                 dbg(x, lb, "case 1");
  105.                 cout << cmap[pref[lb]] << ' ';
  106.             } else if (acc > 0) {
  107.                 cout << find(x) << ' ';
  108.                 dbg(x, lb, find(x), "case 2");
  109.             } else {
  110.                 dbg(x, lb, "case 3");
  111.                 cout << -1 << ' ';
  112.             }
  113.         }
  114.         cout << '\n';
  115.     }
  116.     return 0;
  117. };
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement