题意
给出一张完全图,其中
A
A
A控制着一些城市,
B
B
B也控制着一些城市,它们两之间不重复。
一个城市只有一个导弹可以去摧毁城市,求出
A
A
A要摧毁
B
B
B所有城市至少需要的时间是多少。
思路
显然答案为 A A A的一些城市到 B B B中最长的路最短。
那么根据这个性质,我们可以二分一下时间,使得当前图中所有边都小于它,然后利用最大匹配判断是否能匹配完 B B B。
代码
#include<cstdio>
#include<cstring>
#include<algorithm>
struct node{
int to, next;
}e[100001];
int n, l, r, lengthA, lengthB, tot, ans;
int A[101], B[101], f[101][101], line[101], head[101], v[101];
void add(int u, int v) {
tot++;
e[tot] = (node){v, head[u]};
head[u] = tot;
}
int find(int x) {
for (int i = head[x]; i; i = e[i].next) {
int y = e[i].to;
if (!v[y]) {
v[y] = 1;
if (!line[y] || find(line[y])) {
line[y] = x;
return 1;
}
}
}
return 0;
}
int check(int x) {
tot = 0;
memset(head, 0, sizeof(head));
memset(line, 0, sizeof(line));
for (int i = 1; i <= lengthA; i++)
for (int j = 1; j <= lengthB; j++)
if (f[A[i]][B[j]] && f[A[i]][B[j]] <= x) add(A[i], B[j]);
int flag = 0;
for (int i = 1; i <= n; i++) {
memset(v, 0, sizeof(v));
flag += find(i);
}
return flag == lengthB;
}
int main() {
scanf("%d", &n);
for (int i = 1; i <= n; i++)
for (int j = 1; j <= n; j++)
scanf("%d", &f[i][j]);
for (int k = 1; k <= n; k++)
for (int i = 1; i <= n; i++)
for (int j = 1; j <= n; j++) {
f[i][j] = std::min(f[i][j], f[i][k] + f[k][j]);
r = std::max(r, f[i][j]);
}
scanf("%d", &lengthA);
for (int i = 1; i <= lengthA; i++)
scanf("%d", &A[i]);
scanf("%d", &lengthB);
for (int i = 1; i <= lengthB; i++)
scanf("%d", &B[i]);
int mid;
while (l < r) {
mid = (l + r) / 2;
if (check(mid)) r = mid;
else l = mid + 1;
}
printf("%d", r);
}