oj.uz

Submission #174711

# Submission time Handle Problem Language Result Execution time Memory
174711 2020-01-06T20:59:04 Z stefdasca Museum (CEOI17_museum) C++14
20 / 100
 3000 ms 95856 KB 
#include<bits/stdc++.h>
#define god dimasi5eks
#pragma GCC optimize("O3")
#define fi first
#define se second
#define pb push_back
#define pf push_front
#define mod 1000000007
#define dancila 3.14159265359
#define eps 1e-9

// #define fisier

using namespace std;

typedef long long ll;


int add(int a, int b)
{
    ll x = a+b;
    if(x >= mod)
        x -= mod;
    if(x < 0)
        x += mod;
    return x;
}
ll mul(ll a, ll b)
{
    return (a*b) % mod;
}

ll pw(ll a, ll b)
{
    ll ans = 1;
    while(b)
    {
        if(b & 1)
            ans = (ans * a) % mod;
        a = (a * a) % mod;
        b >>= 1;
    }
    return ans;
}
int n, k, x;
int dp[2][10002][10002];
int xtr[10002][10002];

vector<pair<int, int> >v[10002];

ll dp2[2][10002];
int xtr2[2][10002];
bool viz2[2][10002];
int sts[10002];
void dfs(int dad, int nod, int cost)
{
    sts[nod] = 1;
    for(int i = 0; i < v[nod].size(); ++i)
    {
        int vecin = v[nod][i].fi;
        if(vecin == dad)
            continue;
        dfs(nod, vecin, v[nod][i].se);
    }
    memset(dp2, 0, sizeof(dp2));
    memset(viz2, 0, sizeof(viz2));
    memset(xtr2, 0, sizeof(xtr2));
    viz2[0][1] = 1;
    for(int i = 0; i < v[nod].size(); ++i)
    {
        int vecin = v[nod][i].fi;
        if(vecin == dad)
            continue;
        sts[nod] += sts[vecin];
        int edcost = v[nod][i].se;
        for(int j = sts[nod]; j >= 0; --j)
            for(int trb = 0; trb <= 1; ++trb)
                for(int sz = 1; sz <= min(j, sts[vecin]); ++sz)
                {
                    if(!viz2[trb][j - sz])
                        continue;
                    if(!viz2[trb][j] || dp2[trb][j - sz] + dp[0][vecin][sz] + 2 * edcost < dp2[trb][j])
                    {
                        dp2[trb][j] = dp2[trb][j - sz] + dp[0][vecin][sz] + 2 * edcost;
                        viz2[trb][j] = 1;
                        xtr2[trb][j] = xtr2[trb][j - sz];
                    }
                    else
                        if(dp2[trb][j - sz] + dp[0][vecin][sz] + 2 * edcost == dp2[trb][j])
                            xtr2[trb][j] = min(xtr2[trb][j], xtr2[trb][j - sz]);
                    if(trb == 0)
                    {
                        if(!viz2[1][j] || dp2[0][j - sz] + dp[1][vecin][sz] + edcost < dp2[1][j])
                        {
                            dp2[1][j] = dp2[0][j - sz] + dp[1][vecin][sz] + edcost;
                            viz2[1][j] = 1;
                            xtr2[1][j] = xtr[vecin][sz];
                        }
                        else
                            if(dp2[0][j - sz] + dp[1][vecin][sz] + edcost == dp2[1][j])
                                xtr2[1][j] = min(xtr2[1][j], xtr[vecin][sz]);
                    }
                }
    }
    for(int i = 2; i <= sts[nod]; ++i)
    {
        dp[1][nod][i] = dp2[1][i];
        xtr[nod][i] = xtr2[1][i];
        dp[0][nod][i] = dp2[0][i];
    }
    for(int i = 1; i <= sts[nod]; ++i)
        xtr[nod][i] += cost;
    /*
    cout << nod << '\n';
    for(int i = 1; i <= sts[nod]; ++i)
        cout << dp[0][nod][i] << " ";
    cout << '\n';
    for(int i = 1; i <= sts[nod]; ++i)
        cout << dp[1][nod][i] << " ";
    cout << '\n';
    for(int i = 1; i <= sts[nod]; ++i)
        cout << xtr[nod][i] << " ";
    cout << '\n';
    */
}
int main()
{

    #ifdef fisier
        ifstream f("input.in");
        ofstream g("output.out");
    #endif

    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    cin >> n >> k >> x;
    for(int i = 1; i < n; ++i)
    {
        int a, b, c;
        cin >> a >> b >> c;
        v[a].pb({b, c});
        v[b].pb({a, c});
    }
    dfs(0, x, 0);
    cout << dp[1][x][k] << '\n';
    return 0;
}

Compilation message

museum.cpp: In function 'void dfs(int, int, int)':
museum.cpp:58:22: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     for(int i = 0; i < v[nod].size(); ++i)
                    ~~^~~~~~~~~~~~~~~
museum.cpp:69:22: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     for(int i = 0; i < v[nod].size(); ++i)
                    ~~^~~~~~~~~~~~~~~

Subtask #1
20.0 / 20.0

# Verdict Execution time Memory Grader output
1 Correct 3 ms 1144 KB Output is correct
2 Correct 3 ms 1020 KB Output is correct
3 Correct 3 ms 1016 KB Output is correct
4 Correct 3 ms 1016 KB Output is correct
5 Correct 3 ms 1020 KB Output is correct

Subtask #2
0 / 25.0

# Verdict Execution time Memory Grader output
1 Correct 1283 ms 95224 KB Output is correct
2 Correct 1706 ms 95856 KB Output is correct
3 Execution timed out 3043 ms 71376 KB Time limit exceeded
4 Halted 0 ms 0 KB -

Subtask #3
0 / 35.0

# Verdict Execution time Memory Grader output
1 Correct 1283 ms 95224 KB Output is correct
2 Correct 1706 ms 95856 KB Output is correct
3 Execution timed out 3043 ms 71376 KB Time limit exceeded
4 Halted 0 ms 0 KB -

Subtask #4
0 / 20.0

# Verdict Execution time Memory Grader output
1 Correct 3 ms 1144 KB Output is correct
2 Correct 3 ms 1020 KB Output is correct
3 Correct 3 ms 1016 KB Output is correct
4 Correct 3 ms 1016 KB Output is correct
5 Correct 3 ms 1020 KB Output is correct
6 Correct 1283 ms 95224 KB Output is correct
7 Correct 1706 ms 95856 KB Output is correct
8 Execution timed out 3043 ms 71376 KB Time limit exceeded
9 Halted 0 ms 0 KB -