CCF 201403-4 无线网络(spfa+dp)

问题描述

　　目前在一个很大的平面房间里有 n 个无线路由器,每个无线路由器都固定在某个点上。任何两个无线路由器只要距离不超过 r 就能互相建立网络连接。
　　除此以外,另有 m 个可以摆放无线路由器的位置。你可以在这些位置中选择至多 k 个增设新的路由器。
　　你的目标是使得第 1 个路由器和第 2 个路由器之间的网络连接经过尽量少的中转路由器。请问在最优方案下中转路由器的最少个数是多少?

输入格式

　　第一行包含四个正整数 n,m,k,r。(2 ≤ n ≤ 100,1 ≤ k ≤ m ≤ 100, 1 ≤ r ≤ 10 ⁸)。
　　接下来 n 行,每行包含两个整数 x _i 和 y _i,表示一个已经放置好的无线 路由器在 (x _i, y _i) 点处。输入数据保证第 1 和第 2 个路由器在仅有这 n 个路由器的情况下已经可以互相连接(经过一系列的中转路由器)。
　　接下来 m 行,每行包含两个整数 x _i 和 y _i,表示 (x _i, y _i) 点处可以增设 一个路由器。
　　输入中所有的坐标的绝对值不超过 10 ⁸,保证输入中的坐标各不相同。

输出格式

　　输出只有一个数,即在指定的位置中增设 k 个路由器后,从第 1 个路 由器到第 2 个路由器最少经过的中转路由器的个数。

样例输入

5 3 1 3
0 0
5 5
0 3
0 5
3 5
3 3
4 4
3 0

样例输出

2

思路：由于这一题要选择最多k个地方放置路由器，我们不能直接去求得最短路的值，然后我想到了dp，用dp[i][j]表示第i个节点已经放置了j个路由器的最短路径（代码里写的dis[i][j]），最后推一下转移关系就好，没啥难度，但这一题卡坐标精度…判断两点是否连通会爆int，因此得强转一下成doubleT_T。

#include <iostream>
#include <stdio.h>
#include <stdlib.h>
#include<string.h>
#include<algorithm>
#include<math.h>
#include<queue>
using namespace std;
typedef long long ll;
const int oo=1e9;
struct data
{
    int x,y;
} a[110];
int n,m,k,r;
bool ok(int i,int j)
{
    if(sqrt((double)(a[i].x-a[j].x)*(double)(a[i].x-a[j].x)+(double)(a[i].y-a[j].y)*(double)(a[i].y-a[j].y))-(double)r<1e-6)
        return 1;
    return 0;
}
vector<int>tu[210];
int dis[210][210];///第i个节点用j个新节点的最短路
bool vis[210]= {0};
int main()
{
    cin>>n>>m>>k>>r;
    for(int i=0; i<n; i++)
        cin>>a[i].x>>a[i].y;
    for(int i=n; i<n+m; i++)
        cin>>a[i].x>>a[i].y;
    for(int i=0; i<n+m; i++)
        for(int j=i+1; j<n+m; j++)
            if(ok(i,j))
            {
                tu[i].push_back(j);
                tu[j].push_back(i);
            }
    for(int i=0; i<n+m; i++)
        for(int j=0; j<=k; j++)
            dis[i][j]=oo;
    queue<int>q;
    q.push(0);
    dis[0][0]=0;
    vis[0]=1;
    while(!q.empty())
    {
        int t=q.front();
        q.pop();
        for(int i=0,l=tu[t].size(); i<l; i++)
        {
            int to=tu[t][i];
            if(to<n)
            {
                int flag=0;
                for(int i=0; i<=k; i++)
                    if(dis[t][i]<oo&&dis[t][i]+1<dis[to][i])
                    {
                        flag=1;
                        dis[to][i]=dis[t][i]+1;
                    }
                if(flag&&!vis[to])
                {
                    vis[to]=1;
                    q.push(to);
                }
            }
            else
            {
                int flag=0;
                for(int i=0; i<k; i++)
                    if(dis[t][i]<oo&&dis[t][i]+1<dis[to][i+1])
                    {
                        flag=1;
                        dis[to][i+1]=dis[t][i]+1;
                    }
                if(flag&&!vis[to])
                {
                    vis[to]=1;
                    q.push(to);
                }
            }
        }
    }
    int ans=dis[1][0];
    for(int i=1; i<=k; i++)
        if(dis[1][i]<oo)
            ans=min(ans,dis[1][i]);
    cout<<ans-1<<endl;
    return 0;
}