题目连接:https://ac.nowcoder.com/acm/contest/890/F
题意:给你一个二维平面,平面上有一些气球,让你最多打六枪,横三枪,竖三枪,问最多打多少个球
最开始的思路一定是枚举列(或行),然后对于三行(或列)最多能打多少球,由于打完了行,会对列产生影响,考虑复杂度,想到了用multiset
对每一列的更新变成了log,
就是把间距为r的连续的3列的气球数存起来,并标记以下行的位置,用中间那一列的下标存
这样枚举的列,对于他们影响的行用multiset 更新,结果一定是三列的气球数,加上multiset里更新完之后最大的那个数
#include <bits/stdc++.h>
using namespace std;
const int maxn = 1e5 + 10;
const int M = 1e5;
int cnt[maxn];
vector <int> h[maxn];
multiset<int> s;
void add(int x) {
auto p = s.find(cnt[x]);
s.erase(p);
cnt[x]++;
s.insert(cnt[x]);
}
void del(int x) {
auto p = s.find(cnt[x]);
s.erase(p);
cnt[x]--;
s.insert(cnt[x]);
}
int main()
{
int n, r,x, y, ans = 0;
scanf("%d%d", &n, &r);
for(int i = 0; i < n; i++) {
scanf("%d%d", &x, &y);
h[x].push_back(y);
if(x - r >= 0) h[x - r].push_back(y);
if(x + r <= M) h[x + r].push_back(y);
cnt[y]++;
if(y - r >= 0) cnt[y - r]++;
if(y + r <= M) cnt[y + r]++;
}
for(int i = 0; i <= M; i++) s.insert(cnt[i]) ;//往multiset里添加行的气球数
for(int i = 0; i <= M; i++) {
int res = (int)h[i].size();//三列的气球总数
for(int j = 0; j < h[i].size(); j++) {//如果影响了三行,就更新
del(h[i][j]);
if(h[i][j] - r >= 0) del(h[i][j] - r);
if(h[i][j] + r <= M) del(h[i][j] + r);
}
ans = max(ans, res + (int)(*s.rbegin()));//取最大值
for(int j = 0; j < h[i].size(); j++) {//再更新回去
add(h[i][j]);
if(h[i][j] - r >= 0) add(h[i][j] - r);
if(h[i][j] + r <= M) add(h[i][j] + r);
}
}
printf("%d\n", ans);
return 0;
}