oj.uz

Submission #580895

# Submission time Handle Problem Language Result Execution time Memory
580895 2022-06-22T05:31:48 Z 조영욱(#8362) Beads and wires (APIO14_beads) C++17
0 / 100
 7 ms 9708 KB 
#include <bits/stdc++.h>
using namespace std;

int n;
typedef pair<int,int> P;
vector<P> adj[200001];
long long dp[200001][2];
vector<P> son[200001];
const long long nnn=-1e16;
const long long nn=-1e12-7;

void dfs(int v,int prev) {
    for(int i=0;i<adj[v].size();i++) {
        P nt=adj[v][i];
        if (nt.first!=prev) {
            son[v].push_back(nt);
            dfs(nt.first,v);
        }
    }
}

long long ans(int v,int type) {
    if (dp[v][type]!=nnn) {
        return dp[v][type];
    }
    if (type==0) {
        long long mx1=nn;
        long long mx2=nn;
        long long ret=0;
        for(int i=0;i<son[v].size();i++) {
            ret+=max(ans(son[v][i].first,0),ans(son[v][i].first,1)+son[v][i].second);
            if (mx1<son[v][i].second+ans(son[v][i].first,0)-max(ans(son[v][i].first,0),ans(son[v][i].first,1)+son[v][i].second)) {
                mx2=mx1;
                mx1=son[v][i].second+ans(son[v][i].first,0)-max(ans(son[v][i].first,0),ans(son[v][i].first,1)+son[v][i].second);
            }
            else if (mx2<son[v][i].second+ans(son[v][i].first,0)-max(ans(son[v][i].first,0),ans(son[v][i].first,1)+son[v][i].second)) {
                mx2=son[v][i].second+ans(son[v][i].first,0)-max(ans(son[v][i].first,0),ans(son[v][i].first,1)+son[v][i].second);
            }
        }
        if (mx1+mx2>0) {
            ret+=mx1+mx2;
        }
        return dp[v][type]=ret;
    }
    long long mx=nn;
    long long ret=0;
    for(int i=0;i<son[v].size();i++) {
        ret+=max(ans(son[v][i].first,0),ans(son[v][i].first,1)+son[v][i].second);
        if (mx<son[v][i].second+ans(son[v][i].first,0)-max(ans(son[v][i].first,0),ans(son[v][i].first,1)+son[v][i].second)) {
            mx=son[v][i].second+ans(son[v][i].first,0)-max(ans(son[v][i].first,0),ans(son[v][i].first,1)+son[v][i].second);
        }
    }
    return dp[v][type]=ret+mx;
}

int main(void) {
    scanf("%d",&n);
    for(int i=1;i<n;i++) {
        int u,v,w;
        scanf("%d %d %d",&u,&v,&w);
        adj[u].push_back(P(v,w));
        adj[v].push_back(P(u,w));
    }
    for(int i=1;i<=n;i++) {
        dp[i][0]=nnn;
        dp[i][1]=nnn;
    }
    dfs(1,0);
    printf("%lld",ans(1,0));
}

Compilation message

beads.cpp: In function 'void dfs(int, int)':
beads.cpp:13:18: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   13 |     for(int i=0;i<adj[v].size();i++) {
      |                 ~^~~~~~~~~~~~~~
beads.cpp: In function 'long long int ans(int, int)':
beads.cpp:30:22: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   30 |         for(int i=0;i<son[v].size();i++) {
      |                     ~^~~~~~~~~~~~~~
beads.cpp:47:18: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   47 |     for(int i=0;i<son[v].size();i++) {
      |                 ~^~~~~~~~~~~~~~
beads.cpp: In function 'int main()':
beads.cpp:57:10: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   57 |     scanf("%d",&n);
      |     ~~~~~^~~~~~~~~
beads.cpp:60:14: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   60 |         scanf("%d %d %d",&u,&v,&w);
      |         ~~~~~^~~~~~~~~~~~~~~~~~~~~

Subtask #1
0 / 13.0

# Verdict Execution time Memory Grader output
1 Correct 7 ms 9684 KB Output is correct
2 Correct 6 ms 9684 KB Output is correct
3 Correct 6 ms 9684 KB Output is correct
4 Correct 6 ms 9704 KB Output is correct
5 Correct 5 ms 9684 KB Output is correct
6 Incorrect 5 ms 9708 KB Output isn't correct
7 Halted 0 ms 0 KB -

Subtask #2
0 / 15.0

# Verdict Execution time Memory Grader output
1 Correct 7 ms 9684 KB Output is correct
2 Correct 6 ms 9684 KB Output is correct
3 Correct 6 ms 9684 KB Output is correct
4 Correct 6 ms 9704 KB Output is correct
5 Correct 5 ms 9684 KB Output is correct
6 Incorrect 5 ms 9708 KB Output isn't correct
7 Halted 0 ms 0 KB -

Subtask #3
0 / 29.0

# Verdict Execution time Memory Grader output
1 Correct 7 ms 9684 KB Output is correct
2 Correct 6 ms 9684 KB Output is correct
3 Correct 6 ms 9684 KB Output is correct
4 Correct 6 ms 9704 KB Output is correct
5 Correct 5 ms 9684 KB Output is correct
6 Incorrect 5 ms 9708 KB Output isn't correct
7 Halted 0 ms 0 KB -

Subtask #4
0 / 43.0

# Verdict Execution time Memory Grader output
1 Correct 7 ms 9684 KB Output is correct
2 Correct 6 ms 9684 KB Output is correct
3 Correct 6 ms 9684 KB Output is correct
4 Correct 6 ms 9704 KB Output is correct
5 Correct 5 ms 9684 KB Output is correct
6 Incorrect 5 ms 9708 KB Output isn't correct
7 Halted 0 ms 0 KB -