【链接】 我是链接,点我呀:)
【题意】
给你一棵树
让你选择若干个修理点.
这些修理点被选中之后,节点i到1号节点之间的所有"坏路"都会被修好
问最少需要选择多少个点才能将所有的坏路修好
【题解】
以1为根节点建立一棵树
对于每一棵子树,如果里面有"坏路",必然要选择一个修理点,怎么选择呢?
肯定是"最底下"的那条坏路的下面那个点.
因此记录下每个子树下面的修理点的个数(我们是贪心选择的,能统计出来)
然后如果i和直系儿子j之间有一条坏路,且j所在的子树下面没有修理点,那么这条边肯定只能由一个放在j的修理点修理.
所以在j放一个修理点。
因为是递归的,所以是从下往上,因此满足"最底下"这个条件.
当然i可能有其他的子树,其他的子树的话也按照这样的规则处理就好。
一定要统计这个修理点个数,不然你不知道这条"坏边"是不是所在子树最底下的边(中间的边会被下面的修理点覆盖到不用重新加修理点)
【代码】
import java.io.*;
import java.util.*;
public class Main {
static InputReader in;
static PrintWriter out;
public static void main(String[] args) throws IOException{
//InputStream ins = new FileInputStream("E:\\rush.txt");
InputStream ins = System.in;
in = new InputReader(ins);
out = new PrintWriter(System.out);
//code start from here
new Task().solve(in, out);
out.close();
}
static int N = (int)1e5;
static class Task{
ArrayList<Integer> g[] = new ArrayList[N+10];
ArrayList<Integer> t[] = new ArrayList[N+10];
ArrayList<Integer> ans = new ArrayList();
int n;
int dfs(int x,int pre) {
int len = g[x].size();
int cnt = 0;
for (int i = 0;i < len;i++) {
int y = g[x].get(i),tt = t[x].get(i);
if (y==pre) continue;
int kk = dfs(y,x);
cnt+=kk;
if (kk==0 && tt==2) {
cnt++;
ans.add(y);
}
}
return cnt;
}
public void solve(InputReader in,PrintWriter out) {
for (int i = 1;i <= N;i++) {
g[i] = new ArrayList<Integer>();
t[i] = new ArrayList<Integer>();
}
n = in.nextInt();
for (int i = 1;i <= n-1;i++) {
int x,y,tt;
x = in.nextInt();y = in.nextInt();tt = in.nextInt();
g[x].add(y);t[x].add(tt);
g[y].add(x);t[y].add(tt);
}
dfs(1,0);
out.println(ans.size());
for (int i = 0;i < (int)ans.size();i++) {
out.print(ans.get(i)+" ");
}
}
}
static class InputReader{
public BufferedReader br;
public StringTokenizer tokenizer;
public InputReader(InputStream ins) {
br = new BufferedReader(new InputStreamReader(ins));
tokenizer = null;
}
public String next(){
while (tokenizer==null || !tokenizer.hasMoreTokens()) {
try {
tokenizer = new StringTokenizer(br.readLine());
}catch(IOException e) {
throw new RuntimeException(e);
}
}
return tokenizer.nextToken();
}
public int nextInt() {
return Integer.parseInt(next());
}
}
}