#include <cstdio> #include <iostream> #include <vector> #include <queue> #include <stack> #include <stdlib.h> #include <string.h> #include <stdlib.h> #include <list> #include <algorithm> #include <set> #include <unordered_map> #include <math.h> #define FOR(x,y,z) for(int x = (y); x < (z); x++) #define FORD(x,y,z) for(int x = (y); x >= z; x--) #define REP(r,n) for(int r = 0; r < (n); r++) #define MP make_pair #define ST first #define ND second #define PB push_back #define MAXUS 202000 #define PI 3.1415926 #define epsilon 0.000001 using namespace std; typedef long long LL; typedef long double LD; typedef vector<int> VI; typedef pair<int, int> PR; #define SIZE(c) ((int)((c).size())) #define FOREACH(i,x) for (__typeof((x).begin()) i=(x).begin(); i!=(x).end(); i++) #define ALL(u) (u).begin(),(u).end() using namespace std; bool comp(const PR &a, const PR &b) { return a.first < b.first || (a.first == b.first && a.second < b.second); } int main() { int n, m; // n - number of clients, m - number of bakeries cin >> n >> m; int arrive[n]; LL presum_t_diff[n+1]; vector<int> time_differences(n); vector<int> baking(m); int lastTime = 0; REP(i,n) { cin >> arrive[i]; time_differences[i] = arrive[i] - lastTime; lastTime = arrive[i]; } REP(i,m) {cin >> baking[i];} sort(time_differences.begin(), time_differences.end()); presum_t_diff[0] = time_differences[0]; FOR(i,1,n) { presum_t_diff[i] = presum_t_diff[i-1] + time_differences[i]; } REP(i,m) { LL bake_time = (LL) baking[i]; auto t = upper_bound(time_differences.begin(), time_differences.end(), bake_time); int index = t - time_differences.begin(); LL result = 0; if (index > 0) { result += index * bake_time - presum_t_diff[index-1]; } cout << result << 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 | #include <cstdio> #include <iostream> #include <vector> #include <queue> #include <stack> #include <stdlib.h> #include <string.h> #include <stdlib.h> #include <list> #include <algorithm> #include <set> #include <unordered_map> #include <math.h> #define FOR(x,y,z) for(int x = (y); x < (z); x++) #define FORD(x,y,z) for(int x = (y); x >= z; x--) #define REP(r,n) for(int r = 0; r < (n); r++) #define MP make_pair #define ST first #define ND second #define PB push_back #define MAXUS 202000 #define PI 3.1415926 #define epsilon 0.000001 using namespace std; typedef long long LL; typedef long double LD; typedef vector<int> VI; typedef pair<int, int> PR; #define SIZE(c) ((int)((c).size())) #define FOREACH(i,x) for (__typeof((x).begin()) i=(x).begin(); i!=(x).end(); i++) #define ALL(u) (u).begin(),(u).end() using namespace std; bool comp(const PR &a, const PR &b) { return a.first < b.first || (a.first == b.first && a.second < b.second); } int main() { int n, m; // n - number of clients, m - number of bakeries cin >> n >> m; int arrive[n]; LL presum_t_diff[n+1]; vector<int> time_differences(n); vector<int> baking(m); int lastTime = 0; REP(i,n) { cin >> arrive[i]; time_differences[i] = arrive[i] - lastTime; lastTime = arrive[i]; } REP(i,m) {cin >> baking[i];} sort(time_differences.begin(), time_differences.end()); presum_t_diff[0] = time_differences[0]; FOR(i,1,n) { presum_t_diff[i] = presum_t_diff[i-1] + time_differences[i]; } REP(i,m) { LL bake_time = (LL) baking[i]; auto t = upper_bound(time_differences.begin(), time_differences.end(), bake_time); int index = t - time_differences.begin(); LL result = 0; if (index > 0) { result += index * bake_time - presum_t_diff[index-1]; } cout << result << endl; } return 0; } |