Loading

K 蹦蹦炸弹(大模拟) 哈理工程序设计竞赛

题目:

​ 出题人在\(x\)轴上放置了\(n\)个正在移动的炸弹,第\(i\)个炸弹的初始位置为\(x[i]\),速度为\(v[i]\),当两颗炸弹相遇时会发生爆炸,导致这两颗炸弹消失。在经历了\(10^{100000}\)秒后,出题人想知道最后还剩下几颗炸弹,以及它们的编号。(数据保证不会有三个及以上的炸弹同时相遇)

分析:

​ 由于炸弹运动时间可以看做无限长,所以所有有相对运动迹象的炸弹都会爆炸,我们可以将所有相邻的炸弹的相遇时间扔进优先队列(对于炸弹的3*3种情况讨论),用set维护下标删除炸弹即可。

实现:

#include <bits/stdc++.h>
    
using namespace std;
#define rep(i, a, n) for(int i = a; i < n; i++)
#define all(x) x.begin(), x.end()
#define pb push_back
#define ios ios::sync_with_stdio(false);cin.tie(0);
#define debug(x)    cout << x << endl;
#define SZ(x)    (int)x.size()
typedef long long LL;
typedef unsigned long long ULL;
typedef pair<int, int> PII;
const double inf = 1e18;
void read(int &x) {int s = 0, f = 1; char ch = getchar(); while(!isdigit(ch)) {f = (ch == '-' ? -1 : f); ch = getchar();} while(isdigit(ch)) {s = s * 10 + ch - '0'; ch = getchar();} x = s * f;}

const int N = 100005;
int n;
int vis[N];
struct node {
    int x, v, id;
    bool operator < (const node& aa) const {
        return x < aa.x;
    }
} a[N];
struct Node { //优先队列维护碰撞时间最短的两个炸弹
    int u, v;
    double time;
    bool operator < (const Node& aa) const {
        return time > aa.time;
    }
};
priority_queue<Node> q;
set<int> s;

double calc(int idx1, int idx2) //计算相邻的炸弹的碰撞时间
{
    double dis = abs(a[idx1].x - a[idx2].x) * 1.0;
    int v1 = a[idx1].v, v2 = a[idx2].v;
    if(v1 == v2) //都为0 或者速度相同
        return inf;
    if(v1 == 0)
    {
        if(v2 < 0)
            return dis / abs(v2);
        else
            return inf;
    }
    if(v2 == 0)
    {
        if(v1 > 0)
            return dis / abs(v1);
        else 
            return inf;
    }

    if(v1 < 0 && v2 > 0)
        return inf;
    if(v1 > 0 && v2 < 0)
        return dis / (abs(v1) + abs(v2));
    
    if(v1 < 0 && v2 < 0)
    {
        if(abs(v2) > abs(v1))
            return dis / (abs(v2) - abs(v1));
        else   
            return inf;
    }
    if(v1 > 0 && v2 > 0)
    {
        if(abs(v1) > abs(v2))
            return dis / (abs(v1) - abs(v2));
        else 
            return inf;
    }
    return inf;
}

signed main()
{
    cin >> n;
    for(int i = 0; i <= n + 1; i ++)
        s.insert(i);

    rep(i, 1, n + 1)
    {
        cin >> a[i].x;
        a[i].id = i; 
    }
    rep(i, 1, n + 1)
        cin >> a[i].v;
    sort(a + 1, a + n + 1);

    for(int i = 2; i <= n; i ++)
        q.push({i - 1, i, calc(i - 1, i)});

    while(q.size())
    {
        auto tmp = q.top();
        q.pop();

        if(tmp.time == inf) break;
        if(vis[tmp.u] || vis[tmp.v])    continue;
        vis[tmp.u] = 1;
        vis[tmp.v] = 1;
        int l, r;
        auto iter = s.find(tmp.u);
        l = *(--iter);
        iter ++;
        s.erase(iter);
        iter = s.find(tmp.v);
        r = *(++iter);
        iter --;
        s.erase(iter);
        if(l == 0 || r == n + 1)    continue;
        q.push({l, r, calc(l, r)});
    }

    vector<int> res;
    for(int x : s)
        if(x != 0 && x != n + 1)
            res.pb(a[x].id);
    
    sort(all(res));
    cout << SZ(res) << '\n';
    for(int x : res)
        cout << x << '\n';
}
posted @ 2022-09-07 13:21  DM11  阅读(57)  评论(0编辑  收藏  举报