Submission #1011880

#	Time	Username	Problem	Language	Result	Execution time	Memory
1011880		andrei_iorgulescu	Beads and wires (APIO14_beads)	C++14	0 / 100	6 ms	10076 KiB

beads

#include <bits/stdc++.h>

using namespace std;

#define int long long

int inf = 2e9;

int root;
int n;
vector<pair<int,int>> g[200005], f[200005];
pair<int,int> t[200005];
//int dp[200005][2];

void dfs(int nod)
{
    for (auto vecin : g[nod])
    {
        if (vecin.first != t[nod].first)
        {
            f[nod].push_back(vecin);
            t[vecin.first] = {nod,vecin.second};
            dfs(vecin.first);
        }
    }
}

/*
10
4 10 2
1 2 21
1 3 13
6 7 1
7 9 5
2 4 3
2 5 8
1 6 55
6 8 34
*/

bool iau[15];
int ans;
bool nah;
int dp[15][2];

void solve(int nod)
{
    for (auto vecin : f[nod])
        solve(vecin.first);
    if (f[nod].size() == 0)
    {
        if (iau[nod])
            nah = true;
        dp[nod][0] = 0;
        return;
    }
    if (iau[nod])
    {
        vector<vector<int>> d(f[nod].size());
        for (int i = 0; i < f[nod].size(); i++)
            d[i].resize(3);
        for (int i = 0; i < f[nod].size(); i++)
        {
            int fiu = f[nod][i].first,cost = f[nod][i].second;
            if (i == 0)
            {
                d[i][2] = -inf;
                d[i][0] = max(dp[fiu][0],dp[fiu][1]);
                d[i][1] = dp[fiu][0] + cost;
            }
            else
            {
                d[i][0] = d[i - 1][0] + max(dp[fiu][0],dp[fiu][1]);
                d[i][1] = max(d[i - 1][0] + dp[fiu][0] + cost,d[i - 1][1] + max(dp[fiu][0],dp[fiu][1]));
                d[i][2] = max(d[i - 1][1] + dp[fiu][0] + cost,d[i - 1][2] + max(dp[fiu][0],dp[fiu][1]));
            }
        }
        dp[nod][0] = max(d[f[nod].size() - 1][0],d[f[nod].size() - 1][2]);
        dp[nod][1] = d[f[nod].size() - 1][1] + t[nod].second;
    }
    else
    {
        dp[nod][0] = 0;
        for (auto vecin : f[nod])
            dp[nod][0] += max(dp[vecin.first][0],dp[vecin.first][1]);
    }
}

void afis()
{
    nah = false;
    for (int i = 1; i <= n; i++)
        dp[i][0] = dp[i][1] = -inf;
    solve(root);
    if (!nah)
        ans = max(ans,dp[root][0]);
}

void bkt(int pos)
{
    if (pos == n + 1)
    {
        afis();
        return;
    }
    iau[pos] = true;
    bkt(pos + 1);
    iau[pos] = false;
    bkt(pos + 1);
}

signed main()
{
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);
    cin >> n;
    for (int i = 1; i < n; i++)
    {
        int x,y,z;
        cin >> x >> y >> z;
        g[x].push_back({y,z});
        g[y].push_back({x,z});
    }
    if (n == 2)
    {
        cout << 0;
        return 0;
    }
    for (int i = 1; i <= n; i++)
        if (g[i].size() != 1)
            root = i;
    dfs(root);
    bkt(1);
    cout << ans;
    return 0;
}

Compilation message (stderr)

beads.cpp: In function 'void solve(long long int)':
beads.cpp:60:27: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<std::pair<long long int, long long int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   60 |         for (int i = 0; i < f[nod].size(); i++)
      |                         ~~^~~~~~~~~~~~~~~
beads.cpp:62:27: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<std::pair<long long int, long long int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   62 |         for (int i = 0; i < f[nod].size(); i++)
      |                         ~~^~~~~~~~~~~~~~~

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