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());
        }
    }
}
相关推荐
jyyyx的算法博客几秒前
Leetcode 2337 -- 双指针 | 脑筋急转弯
算法·leetcode
SweetCode12 分钟前
裴蜀定理:整数解的奥秘
数据结构·python·线性代数·算法·机器学习
ゞ 正在缓冲99%…25 分钟前
leetcode76.最小覆盖子串
java·算法·leetcode·字符串·双指针·滑动窗口
xuanjiong26 分钟前
纯个人整理,蓝桥杯使用的算法模板day2(0-1背包问题),手打个人理解注释,超全面,且均已验证成功(附带详细手写“模拟流程图”,全网首个
算法·蓝桥杯·动态规划
惊鸿.Jh1 小时前
【滑动窗口】3254. 长度为 K 的子数组的能量值 I
数据结构·算法·leetcode
明灯L1 小时前
《函数基础与内存机制深度剖析:从 return 语句到各类经典编程题详解》
经验分享·python·算法·链表·经典例题
碳基学AI1 小时前
哈尔滨工业大学DeepSeek公开课:探索大模型原理、技术与应用从GPT到DeepSeek|附视频与讲义免费下载方法
大数据·人工智能·python·gpt·算法·语言模型·集成学习
补三补四1 小时前
机器学习-聚类分析算法
人工智能·深度学习·算法·机器学习
独好紫罗兰1 小时前
洛谷题单3-P5718 【深基4.例2】找最小值-python-流程图重构
开发语言·python·算法
正脉科工 CAE仿真1 小时前
基于ANSYS 概率设计和APDL编程的结构可靠性设计分析
人工智能·python·算法