[SCOI2008]城堡

题目描述

在一个国家里，有n个城市（编号为0 到n-1）。这些城市之间有n条双向道

路相连（编号为0 到n-1），其中编号为i的道路连接了城市i和城市ri（一条道

路可以连接一个城市和它自身），长度为di。n 个城市中有m个拥有自己城堡，

可以抵御敌人侵略。如果没有城堡的城市遭受攻击，则离它最近的城堡将派兵前

往救援。

你的任务是在不超过k个没有城堡的城市中建立城堡，使得所有城市中“离

最近城堡的距离”的最大值尽量小。换句话说，若令dist(c)表示城市c的最近城

堡离它的距离，则你的任务是让max{dist(c)}尽量小。

输入数据保证存在方案使得对于每个城市，至少有一个城堡能够到达。

输入输出格式

输入格式：

输入第一行为三个正整数n, m, k。第二行包含n个整数r0,r1,…,rn-1。第三行

包含n 个整数d0,d1,…,dn-1。第四行包含m 个各不相同的0~n-1 之间的整数，分

别为m个城堡所在的城市编号。

输出格式：

输出仅一行，包含一个整数，即max{dist(c)}的最小值。

输入输出样例

输入样例#1： 复制

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

输出样例#1： 复制

2

输入样例#2： 复制

3 1 1
1 2 0
1 2 3
2

输出样例#2： 复制

1

输入样例#3： 复制

3 1 1
1 2 0
1 2 3
2

输出样例#3： 复制

0

输入样例#4： 复制

10 3 3
0 2 0 0 2 2 8 3 8 7
10 9 1 8 1 3 7 2 8 1
3 4 6

输出样例#4： 复制

3

输入样例#5： 复制

2 0 1
1 0
5 10

输出样例#5： 复制

5

说明

100%的数据满足：2<=n<=50, 1<=di<=106, 0<=m<=n-k

先存图,直接用floyd求出最短路

继续二分最大长度mid，对于每个已有城堡的城市，直接去标记其能到达的城市

然后对于不能到达的

我们将其距离不超过枚举的mid的点期望+1，分别在k次中每次找到最大期望的值进行建城堡。

有个玄学：在比较找出最大期望相同时要找编号尽量大的?????

复杂度O(n^3+logd*k*n^2)

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int n,m,k;
long long dis[101][101];
int cnt[101],vis[101],ans,p[101],son[101];
void find(int mid)
{int i,j,Max,maxi;
  memset(cnt,0,sizeof(cnt));
  for (i=1;i<=n;i++)
    if (vis[i]==0)
      {
    for (j=1;j<=n;j++)
      if (dis[i][j]<=mid) 
        cnt[j]++;
      }
  Max=0,maxi=0;
  for (i=1;i<=n;i++)
    if (cnt[i]>=Max)
      {
    Max=cnt[i];
    maxi=i;
      }
  if (maxi==0) return;
  for (i=1;i<=n;i++)
    if (dis[maxi][i]<=mid) vis[i]=1;
}
bool check(int mid)
{int i,j;
  memset(vis,0,sizeof(vis));
  for (i=1;i<=m;i++)
    {
      for (j=1;j<=n;j++)
    if (dis[p[i]][j]<=mid) vis[j]=1;
    }
  for (i=1;i<=k;i++)
    find(mid);
  for (i=1;i<=n;i++)
    if (vis[i]==0) return 0;
  return 1;
}
int main()
{int i,j;
  long long d;
  cin>>n>>m>>k;
  memset(dis,127/2,sizeof(dis));
  for (i=1;i<=n;i++)
    {
      scanf("%d",&son[i]);
      son[i]++;
    }
  int l=0,r=0;
  for (i=1;i<=n;i++)
    {
      scanf("%lld",&d);
      dis[i][son[i]]=min(dis[i][son[i]],d);
      dis[son[i]][i]=min(dis[i][son[i]],d);
      r+=d;
    }
  for (l=1;l<=n;l++)
  for (i=1;i<=n;i++)
    if (i!=l)
    {
      for (j=1;j<=n;j++)
    if (l!=j&&i!=j)
    {
      dis[i][j]=min(dis[i][j],dis[i][l]+dis[l][j]);
    }
    }
  for (i=1;i<=n;i++)
    dis[i][i]=0;
  for (i=1;i<=m;i++)
    scanf("%d",&p[i]),p[i]++;
  l=0;
  while (l<=r)
    {
      int mid=(l+r)/2;
      if (check(mid)) ans=mid,r=mid-1;
      else l=mid+1;
    }
  cout<<ans;
}