USACO 2014 March Contest Bronze
题目:
这是一个炎热的夏日。
懒洋洋的奶牛贝茜想将自己放置在田野中的某个位置,以便可以在短距离内尽可能多地吃到美味的草。
贝茜所在的田野中共有 N 片草地,我们可以将田野视作一个一维数轴。
第 i 片草地中包含 gi 单位的青草,位置坐标为 xi。
不同草地的位置不同。
贝茜想选取田野中的某个点作为她的初始位置(可能是某片草地所在的点)。
只有一片草地与她的初始位置的距离不超过 K 时,贝茜才能吃到那片草地上的草。
如果贝茜选择最佳初始位置,请确定她可以吃到的青草最大数量。
输入格式
第一行包含两个整数 N 和 K。
接下来 N 行,每行描述一片草地,包含两个整数 gi 和 xi。
输出格式
输出如果贝茜选择最佳初始位置,则她可以吃到的青草最大数量。
数据范围
1≤N≤105, 1≤gi≤10000, 0≤xi≤106, 1≤K≤2×106
输入样例:
4 3
4 7
10 15
2 2
5 1
输出样例:
11
样例解释
最佳初始位置选择为 x=4,可以吃到 x=1,x=2,x=7 处的青草。
思路:点是离散的并不是连续的,每次到这个点时,判断一下他能否在我们当前的 滑动窗口 中,在位置x时,可以吃到[x-k,x+k]区间的草,所以我们的窗口长度为2*k
#include <bits/stdc++.h>
#define int long long
#define x first
#define y second
using namespace std;
const int N = 1e5 + 10;
typedef pair<int, int >pr;
pr p[N];//记录位置,和当前位置的草量
int sum[N];//记录草量的前缀和
int n, k;
signed main()
{
ios::sync_with_stdio(0);
cin >> n >> k;
for (int i = 1; i <= n; i++)
{
int a, b;
cin >> a >> b;
p[i] = {b, a};
}
sort(p + 1, p + 1 + n);
for (int i = 1; i <= n; i++)
sum[i] = p[i].y + sum[i - 1]; //按位置排序
//并算出每个位置草量的前缀和
int ans = 0;
for (int i = 1, j = 1; j <= n; j++)
{
//j来表示当前窗口的左端点,i为右端点
while (p[i + 1].x - p[j].x <= 2 * k && i + 1 <= n)
i++;
//找到在窗口内的最后一个值
int s = sum[i] - sum[j - 1];
ans = max(ans, s);
}
cout << ans << endl;
}
通过这样的分析我们会发现,只要我们的数组够大,一维数组就可以同时记录位置和草量,而这道题的x<=1e6,所以我们可以简化代码
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 1000010;
int s[N], n, m;
int main() {
cin >> n >> m;
for(int i = 0; i < n; i ++) {
int a, b;
cin >> a >> b;
s[b + 1] += a;
}
for(int i = 1; i < N; i ++) s[i] += s[i - 1];
int res = -1;
for(int i = 0; i < N; i ++) {
int l = max(1, i - m), r = min(i + m, N - 1);
//在这里可以直接枚举在该点滑动窗口,也就是我们可用的区间
res = max(res, s[r] - s[l - 1]);
}
cout << res;
return 0;
}
但是如果x的数值为<=1e9的话,就还是要用到第一种代码了