【题目链接】:

【题意】

在一棵树上,保证每个点在距离d之内都有一个警察局;

让你删掉最多的边,使得剩下的森林仍然满足上述约束;

【题解】

从每个警察局开始进行bfs;

这样每个路线第一次到达的点肯定是最短路;

所以bfs的时候遇到没访问过的点,就记录它被访问,即它被这个状态一开始的警察局所管辖(控制);

则可以想象每个警察局都同时往外扩张;

则每个警察局管的范围就确定是那些了;

则那些点肯定是到那个管辖它的警察局最近的;

这样就能确定最后哪些边是要保留下来的了;

删掉剩余的边就好;

(每个点开始找最短路是哪条边的,那条边就保留下来);

警察局不用分配给它边.

【Number Of WA】

0

【完整代码】

#include 
     
      using namespace std;#define lson l,m,rt<<1#define rson m+1,r,rt<<1|1#define LL long long#define rep1(i,a,b) for (int i = a;i <= b;i++)#define rep2(i,a,b) for (int i = a;i >= b;i--)#define mp make_pair#define ps push_back#define fi first#define se second#define rei(x) scanf("%d",&x)#define rel(x) scanf("%lld",&x)#define ref(x) scanf("%lf",&x)#define ms(x,y) memset(x,y,sizeof x)typedef pair
      
        pii;typedef pair
       
         pll;const int dx[9] = {
    0,1,-1,0,0,-1,-1,1,1};const int dy[9] = {
    0,0,0,-1,1,-1,1,-1,1};const double pi = acos(-1.0);const int N = 3e5+1000;int n,k,d,dis[N];int p[N];vector 
        
          G[N];queue 
         
           dl;bool bo[N];int main(){    //freopen("F:\\rush.txt","r",stdin);    ms(dis,255);    rei(n),rei(k),rei(d);    rep1(i,1,k)        rei(p[i]),dl.push(p[i]),dis[p[i]] = 0;    rep1(i,1,n-1)    {        int x,y;        rei(x),rei(y);        G[x].ps(mp(y,i)),G[y].ps(mp(x,i));    }    while (!dl.empty())    {        int x = dl.front();        dl.pop();        for (pii temp:G[x])        {            int y = temp.fi;            if (dis[y]==-1)            {                dis[y] = dis[x]+1;                dl.push(y);            }        }    }    int cnt = 0;    rep1(i,1,n)    if (dis[i]!=0)    {        int mi = 21e8,id = -1;        for (pii temp:G[i])        {            int y = temp.fi;            if (dis[y]