题目链接:https://codeforces.com/problemset/problem/1409/E
题目大意:
给你两个长度为为k的平台,再给你n个小球和每个小球的位置坐标(xi, yi),开始时每个小球开始下落,现在让你任意摆放
这两个平台,问你最多能接住多少球?
输入与输出:
input
4
7 1
1 5 2 3 1 5 4
1 3 6 7 2 5 4
1 1
1000000000
1000000000
5 10
10 7 5 15 8
20 199 192 219 1904
10 10
15 19 8 17 20 10 9 2 10 19
12 13 6 17 1 14 7 9 19 3
output
6
1
5
10
解释:
https://espresso.codeforces.com/8a3aa078020926a84255072419871610d7b6a31f.png
思路:
我们让平台的右端点始终在某个或多个球的正下方(对于任意一种不是这样的放置方法,你都可以在使得答案不变或者更优的情况下让平台变成这种放法)
那么剩下的我们可以dp dp[i][j] 表示当前在位置i(离散化之后的位置),已经放置了j个平台。再用一个变量记录着在位置i放置平台时能接住的小球nums。(我们只要保证更新起点与本次更新没有相交的位置,那么nums就为位置(i-k,i)之间的小球数量)。
代码:
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <map>
using namespace std;
const int maxn = 2e5 + 5;
map<int, int> mm;
int dp[maxn][3];
int main()
{
int t;
scanf("%d", &t);
while (t--) {
int n, k, sm = 0;
scanf("%d%d", &n, &k);
mm.clear();
for (int i = 1; i <= n; i++) {
int x;
scanf("%d", &x);
mm[x]++;
}
for (int i = 1, y; i <= n; i++)
scanf("%d", &y);
int siz = mm.size();
auto it_s = mm.begin();
auto it_c = mm.begin();
for (int i = 1, ii = 0; i <= siz; i++, it_c++) {
sm += it_c->second;
while (it_s->first < it_c->first - k) {
sm -= it_s->second;
ii++;
it_s++;
}
dp[i][0] = dp[i - 1][0];
dp[i][1] = dp[i - 1][1];
dp[i][2] = dp[i - 1][2];
dp[i][2] = max(dp[i][2], dp[ii][1] + sm);
dp[i][1] = max(dp[i][1], dp[ii][0] + sm);
//printf("i=%d 0:%d 1:%d 2:%d\n", i, dp[i][0], dp[i][1], dp[i][2]);
}
//for (int i = 1; i <= siz; i++)
//printf("dp[%d][0]=%d 1:%d 2:%d\n", i, dp[i][0], dp[i][1], dp[i][2]);
printf("%d\n", max(dp[siz][1], dp[siz][2]));
}
return 0;
}