【Codeforces 369C】 Valera and Elections

【链接】 我是链接,点我呀:)
【题意】

给你一棵树
让你选择若干个修理点.
这些修理点被选中之后,节点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());
        }
    }
}