제출 #36585

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
36585		letrongdat	구슬과 끈 (APIO14_beads)	C++14	0 / 100	6 ms	4992 KiB

beads

#include <bits/stdc++.h>

using namespace std;

inline int read()
{
    char c;
    int sign = 1;
    do { c = getchar(); } while (c < '0' || c > '9');
    int res;
    for(res = 0; c >= '0' && c <= '9'; c = getchar())
        res = res * 10 + c -'0';
    return res;
}
#define ii pair < int, int >
#define ft first
#define nd second
#define int long long
#define FOR(i, a, b) for(int i = a; i <= b; ++i)
#define REP(i, a, b) for(int i = a; i  < b; ++i)
const int N = 2e5 +10;
int dp[N][4];
int f[N];
int n;
vector < ii > adj[N];
bool cmp(ii a, ii b)
{
    return a.nd + dp[a.ft][0] - max(dp[a.ft][0], dp[a.ft][1]) > b.nd + dp[b.ft][0] - max(dp[b.ft][0], dp[b.ft][1]);
}
void dfs(int u, int prv = 0, int dd = 0)
{
    FOR(i, 0, 2) dp[u][i] = 0;
    int S = 0;
    REP(i, 0, adj[u].size())
    {
        int v = adj[u][i].first, d = adj[u][i].second;
        if (v == prv) continue;
        dfs(v, u, d);
        dp[u][0] += f[v];
        S += max(dp[v][0], dp[v][1]);
    }
    vector < ii > vt;
    REP(i, 0, adj[u].size())
    {
        int v = adj[u][i].first, d = adj[u][i].second;
        if (v == prv) continue;
        vt.push_back({v, d});
        if (prv != 0)
        {
            dp[u][1] = max(dp[u][1], dd + d + S - max(dp[v][0], dp[v][1]) + dp[v][0]);
        }
    }
    if (vt.size() > 1)
    {
        dp[u][2] = S;
        sort(vt.begin(), vt.end(), cmp);
        FOR(i, 0, 1)
        {
            dp[u][2] += vt[i].nd + dp[vt[i].ft][0] - max(dp[vt[i].ft][0], dp[vt[i].ft][1]);
        }
    }

    FOR(i, 0, 2) f[u] = max(f[u], dp[u][i]);
}
main()
{
   // freopen("beads.inp", "r", stdin);
   // freopen("beads.out", "w", stdout);
    n = read();
    REP(i, 1, n)
    {
        int u, v, d;
        u = read(), v= read(), d = read();
        adj[u].push_back({v, d});
        adj[v].push_back({u, d});
    }
    dfs(1);
    cout << f[1];
}

컴파일 시 표준 에러 (stderr) 메시지

beads.cpp: In function 'int read()':
beads.cpp:8:9: warning: unused variable 'sign' [-Wunused-variable]
     int sign = 1;
         ^~~~
beads.cpp: In function 'void dfs(long long int, long long int, long long int)':
beads.cpp:20:40: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
 #define REP(i, a, b) for(int i = a; i  < b; ++i)
beads.cpp:34:9:
     REP(i, 0, adj[u].size())
         ~~~~~~~~~~~~~~~~~~~             
beads.cpp:34:5: note: in expansion of macro 'REP'
     REP(i, 0, adj[u].size())
     ^~~
beads.cpp:20:40: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
 #define REP(i, a, b) for(int i = a; i  < b; ++i)
beads.cpp:43:9:
     REP(i, 0, adj[u].size())
         ~~~~~~~~~~~~~~~~~~~             
beads.cpp:43:5: note: in expansion of macro 'REP'
     REP(i, 0, adj[u].size())
     ^~~
beads.cpp: At global scope:
beads.cpp:65:6: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
 main()
      ^

• Subtask 1: 0 / 13
• Subtask 2: 0 / 15
• Subtask 3: 0 / 29
• Subtask 4: 0 / 43