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

题目连接:https://ac.nowcoder.com/acm/contest/890/F

题意:给你一个二维平面,平面上有一些气球,让你最多打六枪,横三枪,竖三枪,问最多打多少个球

最开始的思路一定是枚举列(或行),然后对于三行(或列)最多能打多少球,由于打完了行,会对列产生影响,考虑复杂度,想到了用multiset

对每一列的更新变成了log,

就是把间距为r的连续的3列的气球数存起来,并标记以下行的位置,用中间那一列的下标存

这样枚举的列,对于他们影响的行用multiset 更新,结果一定是三列的气球数,加上multiset里更新完之后最大的那个数

 

#include 

using namespace std;
const int maxn = 1e5 + 10;
const int M  = 1e5;

int cnt[maxn];
vector  h[maxn];
multiset 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;
}

 

你可能感兴趣的:(stl,思维,模拟)