Codeforces Round 926 (Div. 2) ---- E. Sasha and the Happy Tree Cutting ----题解

E. Sasha and the Happy Tree Cutting:

题目大意:

思路解析:

现在有一颗树,然后给出了k对路径,然后要求路径上至少有一个结点是被染色了的,如果这k对路径没有共用边,那我们至少需要染色k条边。如果有共用边,那么选择将他染色,那么就可以使多个路径满足条件。现在求最小是染色树。已知最多需要20条边,那我们可以找出哪些边可以进行染色(并且它染色后能解决哪些路径),然后进行状压dp。

代码实现:

import java.util.*;


import java.io.*;
import java.math.*;

public class Main {
    static int[] pre;
    static Vector<int[]>[] g;
    public static void main(String[] args) throws IOException {
        int t = f.nextInt();
        while (t > 0) {
            solve();
            t--;
        }
        w.flush();
        w.close();
    }

    public static void solve() throws IOException{
        int n = f.nextInt();
        g = new Vector[n+1];
        pre = new int[n+1];
        for (int i = 0; i < n + 1; i++) {
            g[i] = new Vector<>();
        }
        for (int i = 0; i < n - 1; i++) {
            int x = f.nextInt();
            int y = f.nextInt();
            g[x].add(new int[] {y, i});
            g[y].add(new int[] {x, i});
        }
        int[] st = new int[n];
        dfs1(1, 0);
        int k = f.nextInt();
        for (int i = 0; i < k; i++) {
            int x = f.nextInt(); int y = f.nextInt();
            if (pre[x] != 0) dfs2(x);

            while ((x ^ y) != 0){
                for (int j = 0; j < g[y].size(); j++) {
                    int[] cur = g[y].get(j);
                    if (cur[0] == pre[y]){
                        y = pre[y];
                        st[cur[1]] |= (1 << i);
                        break;
                    }
                }
            }
        }
        HashSet<Integer> set = new HashSet<>();
        for (int i = 0; i < n; i++) {
            if (st[i] > 0)
                set.add(st[i]);
        }
        int[] dp = new int[1 << k];
        Arrays.fill(dp, 1000);
        dp[0] = 0;
        for (int i = 0; i < (1 << k); i++) {
            for (Integer j: set){
                dp[i | j] = Math.min(dp[i|j], dp[i] + 1);
            }
        }
        w.println(dp[(1 << k) - 1]);
    }

    public static void dfs1(int x, int fa){
        for (int i = 0; i < g[x].size(); i++) {
            int[] cur = g[x].get(i);
            int y = cur[0];
            if (y == fa) continue;
            pre[y] = x;
            dfs1(y, x);
        }

    }

    public static void dfs2(int x){
        int y = pre[x];
        pre[x] = 0;
        while (y != 0){
            int t = pre[y];
            pre[y] = x;
            x = y;
            y = t;
        }
    }



    static PrintWriter w = new PrintWriter(new OutputStreamWriter(System.out));
    static Input f = new Input(System.in);

    static class Input {
        public BufferedReader reader;
        public StringTokenizer tokenizer;

        public Input(InputStream stream) {
            reader = new BufferedReader(new InputStreamReader(stream), 32768);
            tokenizer = null;
        }

        public String next() throws IOException{
            while (tokenizer == null || !tokenizer.hasMoreTokens()) {
                tokenizer = new StringTokenizer(reader.readLine());
            }
            return tokenizer.nextToken();
        }

        public int nextInt() throws IOException {
            return Integer.parseInt(next());
        }

        public long nextLong() throws IOException {
            return Long.parseLong(next());
        }

        public Double nextDouble() throws IOException {
            return Double.parseDouble(next());
        }
    }
}
相关推荐
pianmian14 小时前
python数据结构基础(7)
数据结构·算法
好奇龙猫6 小时前
【学习AI-相关路程-mnist手写数字分类-win-硬件:windows-自我学习AI-实验步骤-全连接神经网络(BPnetwork)-操作流程(3) 】
人工智能·算法
sp_fyf_20247 小时前
计算机前沿技术-人工智能算法-大语言模型-最新研究进展-2024-11-01
人工智能·深度学习·神经网络·算法·机器学习·语言模型·数据挖掘
香菜大丸7 小时前
链表的归并排序
数据结构·算法·链表
jrrz08287 小时前
LeetCode 热题100(七)【链表】(1)
数据结构·c++·算法·leetcode·链表
oliveira-time7 小时前
golang学习2
算法
南宫生8 小时前
贪心算法习题其四【力扣】【算法学习day.21】
学习·算法·leetcode·链表·贪心算法
懒惰才能让科技进步9 小时前
从零学习大模型(十二)-----基于梯度的重要性剪枝(Gradient-based Pruning)
人工智能·深度学习·学习·算法·chatgpt·transformer·剪枝
Ni-Guvara9 小时前
函数对象笔记
c++·算法
泉崎9 小时前
11.7比赛总结
数据结构·算法