【AcWing 158周赛】

5842. 拆树

主要思路:

  • 如何分割?
    深度遍历,当遍历的子数的sum为总sum的三分之一即是一个分割方法。
    1其中要注意节点范围为 [ − 100 , 100 ] [-100,100] [−100,100],所以当 总树 s u m = 0 总树sum=0 总树sum=0时,可能可以分成大于3部分。所以最后输出答案时判断条件应该用 a n s . s i z e ( ) > 2 ans.size() > 2 ans.size()>2
  • 这种方法会不会遗漏答案?
    不会,因为从根节点遍历,任意答案方案都能通过上面的方法得到。
输入样例1:
复制代码
6
2 4
0 5
4 2
2 1
1 1
4 2
输出样例1:
复制代码
1 4
c 复制代码
#include<iostream>
#include<vector>
using namespace std;

const int N = 1e6 + 10;
int d[N], n, h[N], e[N], ne[N], ind = 0;
double sum = 0;
vector<int> ans;

void add(int b, int a) {
    e[++ind] = a;
    ne[ind] = h[b];
    h[b] = ind;
}

void dfs(int r) {
    for (int i = h[r]; i; i = ne[i]) {
        int t = e[i];
        dfs(t);
        d[r] += d[t];
    }
    if (d[r] == sum / 3) {
         ans.push_back(r);
        d[r] = 0;
    }
}
int main() {
     cin >> n;
     int r = 0;
     int x;
     for (int i = 1; i <= n; i++) {
         cin >> x >> d[i];
         if (x) add(x, i);
         else r = i;
         sum += d[i];
     }
    dfs(r);
    if (ans.size() > 2) {
        for (int i =0; i < 2; i++) cout << ans[i] << " ";
    }else puts("-1");
    return 0;
}

5843. 染色

由图可知,所有不连贯的色区边界一定是两色切换的边。

所以,只要遍历整颗树,选两色色块的较小值作为答案。

输入样例1:
复制代码
11
0 0 0 1 1 0 1 0 0 1 1
1 2
1 3
2 4
2 5
5 6
5 7
3 8
3 9
3 10
9 11
输出样例1:
复制代码
2

要注意根节点由于没有父节点,所以和根节点连接的块应该额外加上。

输入

复制代码
15
0 1 0 0 1 1 0 1 1 1 1 1 0 1 0
10 7
10 3
10 8
5 7
13 14
8 13
15 4
15 13
5 2
9 3
11 15
13 6
1 12
9 1

标准答案

复制代码
3
相关推荐
写代码的小球18 分钟前
求模运算符c
算法
大千AI助手4 小时前
DTW模版匹配:弹性对齐的时间序列相似度度量算法
人工智能·算法·机器学习·数据挖掘·模版匹配·dtw模版匹配
YuTaoShao5 小时前
【LeetCode 热题 100】48. 旋转图像——转置+水平翻转
java·算法·leetcode·职场和发展
生态遥感监测笔记5 小时前
GEE利用已有土地利用数据选取样本点并进行分类
人工智能·算法·机器学习·分类·数据挖掘
Tony沈哲6 小时前
macOS 上为 Compose Desktop 构建跨架构图像处理 dylib:OpenCV + libraw + libheif 实践指南
opencv·算法
刘海东刘海东6 小时前
结构型智能科技的关键可行性——信息型智能向结构型智能的转变(修改提纲)
人工智能·算法·机器学习
pumpkin845147 小时前
Rust 调用 C 函数的 FFI
c语言·算法·rust
挺菜的7 小时前
【算法刷题记录(简单题)003】统计大写字母个数(java代码实现)
java·数据结构·算法
mit6.8247 小时前
7.6 优先队列| dijkstra | hash | rust
算法