（CCF 201403-4）无线网络 最短路变形：节点访问限制 + [SPFA]

问题描述
　　目前在一个很大的平面房间里有 n 个无线路由器,每个无线路由器都固定在某个点上。任何两个无线路由器只要距离不超过 r 就能互相建立网络连接。
　　除此以外,另有 m 个可以摆放无线路由器的位置。你可以在这些位置中选择至多 k 个增设新的路由器。
　　你的目标是使得第 1 个路由器和第 2 个路由器之间的网络连接经过尽量少的中转路由器。请问在最优方案下中转路由器的最少个数是多少?
输入格式
　　第一行包含四个正整数 n,m,k,r。(2 ≤ n ≤ 100,1 ≤ k ≤ m ≤ 100, 1 ≤ r ≤ 108)。
　　接下来 n 行,每行包含两个整数 xi 和 yi,表示一个已经放置好的无线 路由器在 (xi, yi) 点处。输入数据保证第 1 和第 2 个路由器在仅有这 n 个路由器的情况下已经可以互相连接(经过一系列的中转路由器)。
　　接下来 m 行,每行包含两个整数 xi 和 yi,表示 (xi, yi) 点处可以增设 一个路由器。
　　输入中所有的坐标的绝对值不超过 108,保证输入中的坐标各不相同。
输出格式
　　输出只有一个数,即在指定的位置中增设 k 个路由器后,从第 1 个路 由器到第 2 个路由器最少经过的中转路由器的个数。
样例输入
5 3 1 3
0 0
5 5
0 3
0 5
3 5
3 3
4 4
3 0
样例输出
2

分析:
这题的模型就是：在一个无向图中，求最短路，不过有一类节点的访问次数是有限制的，即在m个节点中最多只能经过其中k个。但是，整体上还是在求最短路。所以我们还是选择使用SPFA

其次，SPFA最关键处，就是判断在什么情况下可以进入队列，进行迭代优化最短路。
这题是在基础的无向图上加了限制条件，当某一节点可以迭代优化时，我们还要知道目前这条路径，到达该节点时使用了多少限制条件。
不过，只要不超过题目的条件限制数目，那么就可以进入队列进行优化。

我们还是用d[]表示当前最短路的值。
不过，用vis[i][k]:表示当到达i节点，使用了k次条件，来判断队列中是否有重复的状态。

另一种二维求解：
由于设节点maxn时写成了110，结果只过了60%（改成210后就都过了），然后就将状态复杂化了
用d[i][k]来表示到达i节点，使用了k次的最短路。将所有情况铺平开来，这样看起来就比较的清晰了。

一维d[]:

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <queue>
#include <map>
using namespace std;
#define LL long long

const int maxn = 210;
struct node
{
    int id,k;
};

struct edge
{
    int v,next;
}edges[maxn * maxn];

int n,m,k,r,e;
int head[maxn],p[maxn][2];
bool vis[maxn][maxn];
int d[maxn];

void addedges(int u,int v)
{
    edges[e].v = v;
    edges[e].next = head[u];
    head[u] = e++;
    edges[e].v = u;
    edges[e].next = head[v];
    head[v] = e++;
}

void SPFA()
{
    int u,v;
    queue<node> q;
    node now,ne;
    for(int i=0;i<n+m;i++) d[i]= maxn;
    d[0] = 0;
    memset(vis,0,sizeof(vis));
    now.id = 0; now.k = 0;
    q.push(now);
    vis[0][0] = 1;
    while(!q.empty())
    {
        now = q.front(); q.pop();
        vis[now.id][now.k] = 0;
        u = now.id;
        for(int i=head[u];i!=-1;i=edges[i].next)
        {
            v = edges[i].v;
            int kk = now.k;
            if(v >= n) kk++;
            if(d[u] + 1 < d[v] && kk <= k)
            {
                d[v] = d[u] + 1;
                ne.id = v; ne.k = kk;
                if(vis[ne.id][ne.k] == 0)
                {
                    q.push(ne);
                    vis[ne.id][ne.k] = 1;
                }
            }
        }
    }
}


int main()
{
    //freopen("in.txt","r",stdin);
    memset(head,-1,sizeof(head));
    e = 0;
    scanf("%d%d%d%d",&n,&m,&k,&r);
    for(int i=0;i<n+m;i++) scanf("%d%d",&p[i][0],&p[i][1]);
    for(int i=0;i<n+m;i++)
    {
        for(int j=i+1;j<n+m;j++)
        {
            int x = p[i][0] - p[j][0];
            int y = p[i][1] - p[j][1];
            if( (LL)x*x + (LL)y*y <= (LL)r*r ) addedges(i,j);
        }
    }
    SPFA();
    printf("%d\n",d[1]-1);
    return 0;
}

二维d[][]

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <queue>
#include <map>
using namespace std;
#define LL long long

const int maxn = 210;
struct node
{
    int id,k;
};
struct edge
{
    int v,next;
}edges[maxn * maxn];

int n,m,k,r,e;
int head[maxn],p[maxn][2];
bool vis[maxn][maxn];
int d[maxn][maxn];

void addedges(int u,int v)
{
    edges[e].v = v;
    edges[e].next = head[u];
    head[u] = e++;
    edges[e].v = u;
    edges[e].next = head[v];
    head[v] = e++;
}

void SPFA()
{
    int u,v;
    queue<node> q;
    node now,ne;
    for(int i=0;i<n+m;i++)
    {
        for(int j=0;j<n+m;j++)
            d[i][j] = maxn;
    }
    d[0][0] = 0;
    memset(vis,0,sizeof(vis));
    now.id = 0; now.k = 0;
    q.push(now);
    vis[0][0] = 1;
    while(!q.empty())
    {
        now = q.front(); q.pop();
        vis[now.id][now.k] = 0;
        u = now.id;
        for(int i=head[u];i!=-1;i=edges[i].next)
        {
            v = edges[i].v;
            int kk = now.k;
            if(v >= n) kk++;
            if(d[u][now.k] + 1 < d[v][kk] && kk <= k)
            {
                d[v][kk] = d[u][now.k] + 1;
                ne.id = v; ne.k = kk;
                if(vis[ne.id][ne.k] == 0)
                {
                    q.push(ne);
                    vis[ne.id][ne.k] = 1;
                }
            }
        }
    }
}


int main()
{
    //freopen("in.txt","r",stdin);
    memset(head,-1,sizeof(head));
    e = 0;
    scanf("%d%d%d%d",&n,&m,&k,&r);
    for(int i=0;i<n+m;i++) scanf("%d%d",&p[i][0],&p[i][1]);
    for(int i=0;i<n+m;i++)
    {
        for(int j=i+1;j<n+m;j++)
        {
            int x = p[i][0] - p[j][0];
            int y = p[i][1] - p[j][1];
            if( (LL)x*x + (LL)y*y <= (LL)r*r ) addedges(i,j);
        }
    }
    SPFA();
    int ans = maxn;
    for(int i=0;i<=k;i++) ans = min(ans,d[1][i]);
    printf("%d\n",ans - 1);
    return 0;
}