道路重建【洛谷P3905】

题目描述

一场可怕的地震后，人们用N个牲口棚(1≤N≤150，编号1…N)重建了农夫John的牧场。由于人们没有时间建设多余的道路，所以现在从一个牲口棚到另一个牲口棚的道路是惟一的。因此，牧场运输系统可以被构建成一棵树。John想要知道另一次地震会造成多严重的破坏。有些道路一旦被毁坏，就会使一棵含有P(1≤P≤N)个牲口棚的子树和剩余的牲口棚分离，John想知道这些道路的最小数目。
输入输出格式

输入格式：

第1行：2个整数，N和P
第2…N行：每行2个整数I和J，表示节点I是节点J的父节点。

输出格式：

单独一行，包含一旦被破坏将分离出恰含P个节点的子树的道路的最小数目。
输入输出样例
输入样例#1：
6
2
3
4
5
6
7
8
9
10
11
输出样例#1：
说明
【样例解释】
如果道路1-4和1-5被破坏，含有节点（1，2，3，6，7，8）的子树将被分离出来

思路
将完好的道路权值设为0，建图然后跑一遍最短路

#include<bits/stdc++.h>
using namespace std;

int n,m,a,b,d;
int c[118][118];
bool xc[118][118];

int main(){
    scanf("%d%d",&n,&m);
    memset(c,0x3f,sizeof(c));
    for(int i=1;i<=m;i++)
    {
        int u,v,w;
        scanf("%d%d%d",&u,&v,&w);
        xc[u][v]=xc[v][u]=1;
        c[u][v]=c[v][u]=w;
    }
    scanf("%d",&d);
    for(int i=1;i<=d;i++)
    {
        int u,v;
        scanf("%d%d",&u,&v);
        xc[u][v]=xc[v][u]=0;
    }
    for(int i=1;i<=n;i++)
        for(int j=1;j<i;j++)
            if(xc[i][j]) c[i][j]=c[j][i]=0;
    scanf("%d%d",&a,&b);
    for(int k=1;k<=n;k++)
        for(int i=1;i<=n;i++)
            for(int j=1;j<=n;j++)
                if(c[i][k]+c[k][j]<c[i][j]) c[i][j]=c[i][k]+c[k][j];
    printf("%d\n",c[a][b]);
    return 0;
}