2019牛客暑期多校训练营（第十场）F Popping Balloons （multiset + 思维）

题目连接：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;
}