bzoj5438: 青蛙

Posted 2021-01-03 iamqzh233

考虑青蛙的位置始终都是在一个区间里面。

每次是最前面的那只青蛙往下一个没有到达的位置跳。

我们可以求出最多可以有几只青蛙这么跳，把代价最大的扔进去。

如果没有方法可以这么跳，选择代价最小的把所有点跳完，其余的从1跳到n即可。

 1 #include <bits/stdc++.h>
 2 using namespace std;
 3 #define M 100010
 4 #define LL long long
 5 int c[M], a[M];
 6 inline void solve() {
 7     int n, m, k, d;
 8     scanf("%d%d%d%d", &n, &m, &k, &d);
 9     for(int i = 1; i <= m; ++ i) {
10         scanf("%d", &c[i]);
11     }
12     for(int i = 1; i <= k; ++ i) {
13         scanf("%d", &a[i]);
14     }
15     a[++ k] = 1; a[++ k] = n;
16     sort(c + 1, c + m + 1);
17     sort(a + 1, a + k + 1);
18     int lim = 2147483647, r = 1;
19     for(int i = 1; i <= k; ++ i) {
20         r = max(r, i);
21         while(a[r + 1] - a[i] <= d && r < k) ++ r;
22         lim = min(lim, r - i + 1);
23         if(r == k) break;
24     }
25     -- lim;
26     //cerr << "num " << lim << endl;
27     /*for(int i = 1; i <= k; ++ i) {
28         printf("%d ", a[i]);
29     }
30     puts("");*/
31     if(lim == 0) {
32         LL Ans = 0;
33         for(int i = 1; i < k; ++ i) {
34             if(a[i + 1] - a[i] > d) {
35                 Ans += c[1];
36             }
37         }
38         for(int i = 2; i <= m; ++ i) {
39             Ans += c[i];
40         }
41         printf("%lld
", Ans);
42         return;
43     }
44     m -= lim;
45     if(n - 1 <= d) {
46         puts("0");
47         return;
48     }
49     LL Ans = 0;
50     for(int i = 1; i <= m; ++ i) {
51         Ans += c[i];
52     }
53     printf("%lld
", Ans);
54 }
55 int main() {
56     int T;
57     scanf("%d", &T);
58     while(T --) {
59         solve();
60     }
61 }