CF1388D 思维 + dfs

dfs一次解决sum

第二次解决已处理的点的顺序以及未处理的点的顺序

已处理，则存入队列，靠近叶子的点先出

未处理，则存入栈，靠近顶端的点先出

一共跑了三次树

#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<vector>
#define ll long long
using namespace std;
const int N = 2e5 + 10;
int n;
ll a[N], sum[N];
int id[N];

vector<int> v[N];
ll res = 0; 

void dfs(int now)
{
    sum[now] = a[now];
    for(int i = 0 ; i < v[now].size() ; i++){
        int son = v[now][i];
        dfs(son);
        if(sum[son] > 0)    sum[now] += sum[son];
    }
}

void dfs2(int now)
{
    for(int i = 0 ; i < v[now].size() ; i++){
        int son = v[now][i];
        if(sum[son] >= 0)    dfs2(son);
    }
    printf("%d ",now);
    for(int i = 0 ; i < v[now].size() ; i++){
        int son = v[now][i];
        if(sum[son] < 0)    dfs2(son);
    }
}

int main(){
    scanf("%d",&n);
    for(int i = 1 ; i <= n ; i++){
        scanf("%lld",&a[i]);
    }
    for(int i = 1 ; i <= n ; i++){
        int x;scanf("%d",&x);
        if(x != -1){
            v[x].push_back(i);    
            id[i]++;
        }
    }
    
    for(int i = 1 ; i <= n ; i++){
        if(!id[i])    dfs(i);
    }
    for(int i = 1 ; i <= n ; i++){
        res += sum[i];
    }
    
    printf("%lld
",res);
    for(int i = 1 ; i <= n ; i++){
        if(!id[i])    dfs2(i);
    }//从没父亲的节点开始搜叭 
    printf("
");
    return 0;
}

一次dfs解决
每访问一个节点，直接加到res，每个节点至少加一次
若当前节点为正，直接加到它的父亲上（有父亲），同时加到队列里面
若当前节点为负，直接加到栈里面（reverse的vector）

#include <bits/stdc++.h>
#define ll long long
#define all(a) a.begin(),a.end()

using namespace std;

vector <vector <int> > edge;
vector <ll> a;
vector <int> b, used;
vector <int> order[2];
ll ans;
inline void dfs (int v) {
    used[v] = 1;
    for (int to : edge[v]) {
        if (!used[to]) dfs(to);
    }
    ans += a[v];
    if (b[v] != -1 && a[v] > 0) {
        a[b[v]] += a[v];
    }
    if (a[v] > 0) {
        order[0].push_back(v);
    }
    else {
        order[1].push_back(v);
    }
}
inline void solve () {
    for (auto &i : edge) i.clear();
    edge.clear();
    a.clear();
    order[0].clear();
    order[1].clear();
    b.clear();
    used.clear();
    int n, x;
    cin >> n;
    edge.resize(n);
    a.resize(n);
    b.resize(n);
    for (auto &i : a) cin >> i;
    for (int i = 0; i < n; i++) {
        cin >> x;
        if (x != -1) {
            --x;
            edge[x].push_back(i);
        }
        b[i] = x;
    }
    ans = 0;
    used.assign(n, 0);
    for (int i = 0; i < n; i++) {
        if (!used[i]) {
            dfs(i);
        }
    }
    cout << ans << '
';
    reverse(all(order[1]));
    for (auto &i : order[0]) cout << i + 1 << ' ';
    for (auto &i : order[1]) cout << i + 1 << ' ';
    cout << '
';
}
main()
{
    ios::sync_with_stdio(0);
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);
    solve();
}