oj.uz

답안 #799394

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
799394 2023-07-31T13:47:26 Z future Financial Report (JOI21_financial) C++14
0 / 100
 7 ms 10196 KB 
#include <bits/stdc++.h>

using namespace std;
const int N=1e5+5;
int n,m;
vector<int>g[N];
int tab[23][N];
int depth[N];
int pos=0;
int in[N];
int out[N];
void DFS(int u,int par) {
    pos++;
    in[u]=pos;
    for(auto v:g[u])
        if(v!=par) {
            tab[0][v]=u;
            depth[v]=depth[u]+1;
            DFS(v,u);
        }
    out[u]=pos;
}
void init() {
    depth[1]=1;
    DFS(1,0);
    for(int i=1; i<=20; i++)
        for(int j=1; j<=n; j++)
            tab[i][j]=tab[i-1][tab[i-1][j]];
}
int LCA(int u,int v) {
    if(depth[u]<depth[v])swap(u,v);
    for(int i=20; i>=0; i--)if(depth[tab[i][u]]>=depth[v])u=tab[i][u];
    if(u==v)return u;
    for(int i=20; i>=0; i--)if(tab[i][u]!=tab[i][v]) {
            v=tab[i][v];
            u=tab[i][u];
        }
    return tab[0][u];
}
long long dp[N];
struct event {
    int u,v,c;
};
vector<event>save[N];
long long aib[N];
void add(int i, int x) {
    while (i <N) {
        aib[i] += x;
        i += i & (-i);
    }
}

long long get(int i) {
    int sol = 0;
    while (i) {
        sol += aib[i];
        i -= i & (-i);
    }
    return sol;
}
long long sum[N];

void DFS_ans(int u,int par) {
    for(auto v:g[u])
        if(v!=par) {
            DFS_ans(v,u);
            sum[u]+=dp[v];
        }
    dp[u]=sum[u];
    for(auto v:save[u]) {
        dp[u]=max(dp[u],get(in[v.u])+get(in[v.v])+v.c+sum[u]);
    }
    add(in[u],sum[u]-dp[u]);
    add(out[u]+1,dp[u]-sum[u]);
}
int main() {
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);
    cin>>n;
    for(int i=1; i<n; i++) {
        int u,v;
        cin>>u>>v;
        g[u].push_back(v);
        g[v].push_back(u);
    }
    init();
    cin>>m;
    while(m--) {
        int u,v,c;
        cin>>u>>v>>c;
        save[LCA(u,v)].push_back({u,v,c});
    }
    DFS_ans(1,0);
    cout<<dp[1];
    return 0;
}

Subtask #1
0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Runtime error 7 ms 10196 KB Execution killed with signal 11
2 Halted 0 ms 0 KB -

Subtask #2
0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Runtime error 7 ms 10196 KB Execution killed with signal 11
2 Halted 0 ms 0 KB -

Subtask #3
0 / 20.0

# 결과 실행 시간 메모리 Grader output
1 Runtime error 7 ms 10196 KB Execution killed with signal 11
2 Halted 0 ms 0 KB -

Subtask #4
0 / 12.0

# 결과 실행 시간 메모리 Grader output
1 Runtime error 6 ms 9980 KB Execution killed with signal 11
2 Halted 0 ms 0 KB -

Subtask #5
0 / 5.0

# 결과 실행 시간 메모리 Grader output
1 Runtime error 6 ms 9940 KB Execution killed with signal 11
2 Halted 0 ms 0 KB -

Subtask #6
0 / 35.0

# 결과 실행 시간 메모리 Grader output
1 Runtime error 7 ms 10196 KB Execution killed with signal 11
2 Halted 0 ms 0 KB -