HDU 4400-Mines-KDtree

题意就是给n个炸弹, 选择m个去引爆,每个炸弹的影响范围是D,这里的距离是曼哈顿距离,炸弹之间会相互引爆

问每次选择的一个炸弹后,最后引爆的炸弹个数


构建kdtree,然后每次搜最近点,加到队列bfs炸开,复杂度 构建n*logn,查询n*logn



#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
using namespace std;
const int INF = 0x7fffffff; 
struct node
{
    int d[2], l, r,rad,id;
    int Max[2], Min[2];
} t[100000 + 10],tmp;
int ori[100000+10][3];
int vis[100000+500];
int n,m,root,cmp_d;
int k1, k2, k3, Ans;

bool cmp(node a, node b)
{
    return (a.d[cmp_d]= t[p].d[D])
        {
            if (t[p].r == 0)
            {
                t[p].r = now;
                return;
            }
            else p = t[p].r;
        }
        else
        {
            if (t[p].l == 0)
            {
                t[p].l = now;
                return;
            }
            else p = t[p].l;
        }
        D = D ^ 1;
    }
    return;
}
int ABS(int x)
{
    return x < 0? -x : x;
}
int dist(int p1, int px, int py)
{
    int dist = 0;
    if (px < t[p1].Min[0]) dist += t[p1].Min[0] - px;
    if (px > t[p1].Max[0]) dist += px - t[p1].Max[0];
    if (py < t[p1].Min[1]) dist += t[p1].Min[1] - py;
    if (py > t[p1].Max[1]) dist += py - t[p1].Max[1];
    return dist;
}
int sum;
queue q;
void ask(int p)
{
    int dl, dr, d0;
    d0=ABS(t[p].d[0] - k2) + ABS(t[p].d[1] - k3);
    if(d0 <= Ans)
    {
        if(!vis[t[p].id]) q.push(t[p].id);
    }
    if(t[p].l) dl = dist(t[p].l, k2, k3);
    else dl =0x7f7f7f7f;
    if(t[p].r) dr = dist(t[p].r, k2, k3);
    else dr = 0x7f7f7f7f;
    if(dl < dr)
    {
        if(dl <= Ans) ask(t[p].l);
        if(dr <= Ans) ask(t[p].r);
    }
    else
    {
        if(dr <= Ans) ask(t[p].r);
        if(dl <= Ans) ask(t[p].l);
    }
}

void init()
{
    memset(t,0,sizeof t);
    memset(vis,0,sizeof vis);
    for (int i = 1; i <= n; i++)
        scanf("%d%d%d", &t[i].d[0], &t[i].d[1],&t[i].rad);
    for (int i = 1; i <= n; i++)
        t[i].id=i;
    for (int i = 1; i <= n; i++)
        ori[i][0]=t[i].d[0], ori[i][1]=t[i].d[1] , ori[i][2]=t[i].rad;
    root = build(1, n, 0);
    scanf("%d",&m);

}
void work()
{
    for (int i = 1; i <= m ; i++)
    {
        int x;
        scanf("%d",&x);

        sum=0;
        q.push(x);
        while(!q.empty())
        {
            int tp=q.front();
            q.pop();
            if (!vis[tp])
            {
                vis[tp]=1;
                sum++;
                Ans = ori[tp][2];
                k2=ori[tp][0];
                k3=ori[tp][1];
                ask(root);
            }
        }
        printf("%d\n", sum);
    }
}

int main()
{
    int cnt=1;
    while(   scanf("%d", &n )!=EOF&&n)
    {
        printf("Case #%d:\n",cnt++);
        init();
        work();
    }
    return 0;
}


你可能感兴趣的:(数据结构,KD树)