标签：false for bsp str cond 注意 using code stack
题意：n个城市，k个警察局，n-1条边连成树，所有边长都为1，给定的图满足规则：任一城市到离它最近的警察局距离不超过d。 问你最多可以删掉多少条边，使得依旧满足规则。
tags：从所有警察局开始一起bfs，这样当要走向一个点to的时候，肯定是离警察局最近的路。如果to没有走过，就说明这条边 i 是需要的，标记它。最后没有被标记到的边就是不需要的。 注意坑点：d 可以为0
#include<bits/stdc++.h> using namespace std; #pragma comment(linker, "/STACK:102400000,102400000") #define rep(i,a,b) for (int i=a;i<=b;i++) #define per(i,b,a) for (int i=b;i>=a;i--) #define mes(a,b) memset(a,b,sizeof(a)) #define INF 0x3f3f3f3f #define MP make_pair #define PB push_back #define fi first #define se second typedef long long ll; const int N = 300005, M = N<<1; int n, k, d, pli[N], mp[M], ans; bool mark[M], vis[N], vise[M]; struct Edge { int to, next; } e[M]; int tot, head[N]; void Addedge(int u, int v, int x) { e[tot]={v, head[u]}, mp[tot]=x, head[u]=tot++; e[tot]={u, head[v]}, mp[tot]=x, head[v]=tot++; } queue<int > q; int q1[N], sz; void bfs() { rep(ca,1,d) { sz=0; while(!q.empty()) { q1[++sz]=q.front(); q.pop(); } rep(cb,1,sz) { int u=q1[cb]; for(int i=head[u]; i!=-1; i=e[i].next) if(vise[mp[i]]==false) { vise[mp[i]]=true; int to=e[i].to; if(vis[to]==false) { q.push(to), vis[to]=true, mark[mp[i]]=true; } } } } } void Init() { tot=0; ans=0; mes(head, -1); mes(mark, false); mes(vis, false); mes(vise, false); } int main() { scanf("%d %d %d", &n, &k, &d); Init(); rep(i,1,k) { scanf("%d", &pli[i]); vis[pli[i]]=true, q.push(pli[i]); } int u, v; rep(i,1,n-1) { scanf("%d %d", &u, &v); Addedge(u, v, i); } if(d==0) { printf("%d\n", n-1); rep(i,1,n-1) printf("%d ", i); return 0; } bfs(); rep(i,1,n-1) if(mark[i]==false) ++ans; printf("%d\n", ans); rep(i,1,n-1) if(mark[i]==false) printf("%d ", i); return 0; }
标签：false for bsp str cond 注意 using code stack
原文地址：http://www.cnblogs.com/sbfhy/p/7150088.html
