Hdu 4400 Mines - 搜索 + STL

来源：互联网发布：中小学辅导软件编辑：程序博客网时间：2024/05/16 09:23

题目大意：

有点像扫雷游戏，炸弹爆炸的时候就像BFS一样。要确定爆炸的范围。

题目分析：

数据很大，暴力会超时。那么只有离散化以后，在枚举了。

注意枚举的上界和下届，很显然，对于x坐标的下界为x2-x1，上界为x2+x1；对于y坐标的下界为y2-y1,，上界为y2+y1。

下面是代码：

#include <stdio.h>#include <string.h>#include <queue>#include <set>#include <math.h>#include <algorithm>using namespace std;const int maxn = 101000;int n,m,cnt;int X[maxn];int vis[maxn];struct node{    int y;    int id;    node(int a,int b)    {        id = a;        y = b;    }    bool operator < (const node &c) const    {        return y < c.y;    }};struct Mine{    int x;    int y;    int d;} M[maxn];multiset<node> G[maxn];int bfs(int x){    int ans = 0;    if(vis[x]) return ans;    vis[x] = 1;    queue<int> q;    q.push(x);    multiset<node>::iterator yl,yr,it;    while(!q.empty())    {        ans++;        int k = q.front(); q.pop();        int l = lower_bound(X,X+cnt,M[k].x-M[k].d) - X;        int r = upper_bound(X,X+cnt,M[k].x+M[k].d) - X;        //printf("k = %d l = %d r = %d\n",k,l,r);        for(int i = l; i < r; i++)        {            int dy = M[k].d - abs(M[k].x - X[i]);            yl = G[i].lower_bound(node(0,M[k].y-dy));            yr = G[i].upper_bound(node(0,M[k].y+dy));            for(it = yl; it != yr; it++)            {                if(!vis[it->id])                {                    //printf("it->id = %d\n",it->id);                    vis[it->id] = 1;                    q.push(it->id);                }            }            G[i].erase(yl,yr);        }    }    return ans;}int main(){    int T = 0;    while(scanf("%d",&n) && n)    {        cnt = 0;        for(int i = 0; i < n; i++)        {            scanf("%d%d%d",&M[i].x,&M[i].y,&M[i].d);            X[cnt++] = M[i].x;        }        sort(X,X+cnt);        cnt = unique(X,X+cnt) - X;        for(int i = 0; i < cnt; i++) G[i].clear();        for(int i = 0; i < n; i++)        {            int id = lower_bound(X,X+cnt,M[i].x) - X;            G[id].insert(node(i,M[i].y));        }        scanf("%d",&m);        printf("Case #%d:\n",++T);        memset(vis,0,sizeof(vis));        for(int i = 0; i < m; i++)        {            int id;            scanf("%d",&id);            printf("%d\n",bfs(id-1));        }    }    return 0;}