【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
相关推荐
三维重建-光栅投影40 分钟前
VS中将cuda项目编译为DLL并调用
算法
课堂剪切板3 小时前
ch03 部分题目思路
算法
山登绝顶我为峰 3(^v^)34 小时前
如何录制带备注的演示文稿(LaTex Beamer + Pympress)
c++·线性代数·算法·计算机·密码学·音视频·latex
Two_brushes.5 小时前
【算法】宽度优先遍历BFS
算法·leetcode·哈希算法·宽度优先
森焱森7 小时前
水下航行器外形分类详解
c语言·单片机·算法·架构·无人机
QuantumStack9 小时前
【C++ 真题】P1104 生日
开发语言·c++·算法
写个博客10 小时前
暑假算法日记第一天
算法
绿皮的猪猪侠10 小时前
算法笔记上机训练实战指南刷题
笔记·算法·pta·上机·浙大
hie9889410 小时前
MATLAB锂离子电池伪二维(P2D)模型实现
人工智能·算法·matlab
杰克尼11 小时前
BM5 合并k个已排序的链表
数据结构·算法·链表