Advertisement
pb_jiang

CF1490G AC

Jan 12th, 2023
107
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 2.19 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.         // dbg(acc);
  46.  
  47.         ll x;
  48.         for (int i = 0; i < m; ++i) {
  49.             cin >> x;
  50.             int lb = lower_bound(pref.begin(), pref.end(), x) - pref.begin();
  51.             if (lb != pref.size()) {
  52.                 // dbg(x, lb, "case 1");
  53.                 cout << cmap[pref[lb]] << ' ';
  54.             } else if (acc > 0) {
  55.                 ll repeat = (x - pref.back() + acc - 1) / acc;
  56.                 ll delta = x - repeat * acc;
  57.                 ll lb = lower_bound(pref.begin(), pref.end(), delta) - pref.begin();
  58.                 ll ans = repeat * n + cmap[pref[lb]];
  59.                 // dbg(repeat, n, delta, lb, ans);
  60.                 cout << ans << ' ';
  61.             } else {
  62.                 // dbg(x, lb, "case 3");
  63.                 cout << -1 << ' ';
  64.             }
  65.         }
  66.         cout << '\n';
  67.     }
  68.     return 0;
  69. };
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement