#include <cstdio> #include <algorithm> using namespace std; long long t[200001]; pair<int, int> d[200001]; long long ret[200001]; int main() { int n, m; scanf("%d %d", &n, &m); long long aryt = (1LL * n * (n + 1)) / 2; long long sall = 0; for (int i = 0; i < n; i++) { scanf("%lld", t + i); sall += t[i]; } for (int i = 0; i < m; i++) { scanf("%d", &d[i].first); d[i].second = i; } sort(d, d + m); int mj = -1; for (int j = 0; j < m; j++) { long long s = 0; long long last = 0; bool ar = true; for (int i = 0; i < n; i++) { if (last + d[j].first <= t[i] ) { ar = false; last = t[i]; } else { last += d[j].first; } s += last - t[i]; } ret[d[j].second] = s; if (ar) { mj = j + 1; break; } } if (mj > -1) for (int j = mj; j < m; j++) { ret[d[j].second] = aryt * d[j].first - sall; } for (int j = 0; j < m; j++) { printf("%lld\n", ret[j]); } 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 | #include <cstdio> #include <algorithm> using namespace std; long long t[200001]; pair<int, int> d[200001]; long long ret[200001]; int main() { int n, m; scanf("%d %d", &n, &m); long long aryt = (1LL * n * (n + 1)) / 2; long long sall = 0; for (int i = 0; i < n; i++) { scanf("%lld", t + i); sall += t[i]; } for (int i = 0; i < m; i++) { scanf("%d", &d[i].first); d[i].second = i; } sort(d, d + m); int mj = -1; for (int j = 0; j < m; j++) { long long s = 0; long long last = 0; bool ar = true; for (int i = 0; i < n; i++) { if (last + d[j].first <= t[i] ) { ar = false; last = t[i]; } else { last += d[j].first; } s += last - t[i]; } ret[d[j].second] = s; if (ar) { mj = j + 1; break; } } if (mj > -1) for (int j = mj; j < m; j++) { ret[d[j].second] = aryt * d[j].first - sall; } for (int j = 0; j < m; j++) { printf("%lld\n", ret[j]); } return 0; } |