Submission #396848

# Submission time Handle Problem Language Result Execution time Memory
396848 2021-04-30T19:44:55 Z MarcoMeijer Training (IOI07_training) C++14
100 / 100
121 ms 30916 KB
#include <bits/stdc++.h>
using namespace std;
 
// macros
typedef long long ll;
typedef long double ld;
typedef pair<int, int> ii;
typedef pair<ll, ll> lll;
typedef tuple<int, int, int> iii;
typedef vector<int> vi;
typedef vector<ii> vii;
typedef vector<iii> viii;
typedef vector<ll> vll;
typedef vector<lll> vlll;
#define REP(a,b,c) for(int a=int(b); a<int(c); a++)
#define RE(a,c) REP(a,0,c)
#define RE1(a,c) REP(a,1,c+1)
#define REI(a,b,c) REP(a,b,c+1)
#define REV(a,b,c) for(int a=int(c-1); a>=int(b); a--)
#define FOR(a,b) for(auto& a : b)
#define all(a) a.begin(), a.end()
#define INF 1e18
#define EPS 1e-9
#define pb push_back
#define popb pop_back
#define fi first
#define se second
mt19937_64 rng(chrono::steady_clock::now().time_since_epoch().count());
 
// input
template<class T> void IN(T& x) {cin >> x;}
template<class H, class... T> void IN(H& h, T&... t) {IN(h); IN(t...); }
 
// output
template<class T1, class T2> void OUT(const pair<T1,T2>& x);
template<class T> void OUT(const vector<T>& x);
template<class T> void OUT(const T& x) {cout << x;}
template<class H, class... T> void OUT(const H& h, const T&... t) {OUT(h); OUT(t...); }
template<class T1, class T2> void OUT(const pair<T1,T2>& x) {OUT(x.fi,' ',x.se);}
template<class T> void OUT(const vector<T>& x) {RE(i,x.size()) OUT(i==0?"":" ",x[i]);}
template<class... T> void OUTL(const T&... t) {OUT(t..., "\n"); }
template<class H> void OUTLS(const H& h) {OUTL(h); }
template<class H, class... T> void OUTLS(const H& h, const T&... t) {OUT(h,' '); OUTLS(t...); }
 
// dp
template<class T> bool ckmin(T&a, T&b) { bool bl = a > b; a = min(a,b); return bl;}
template<class T> bool ckmax(T&a, T&b) { bool bl = a < b; a = max(a,b); return bl;}
 
void program();
int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);
    program();
}
 
 
//===================//
//   begin program   //
//===================//
 
const int MX = 1050;
const int N = 5005;

int n, m;
int a, b, c;
vi adj[MX];
vi chl[MX];
vi path[N];
vi pathsOnVertex[MX];
int childrenBitMask[MX][N];
int par[MX];
int depth[MX];
int rootPath[N];
int childNumber[MX];
viii unPaved;
int totUnPaved;

void dfsChl(int u, int p) {
    par[u] = p;
    childNumber[u] = -1;
    depth[u] = p == -1 ? 0 : depth[p] + 1;
    int cc = 0;
    FOR(v,adj[u]) if(v != p) {
        dfsChl(v,u);
        childNumber[v] = cc++;
        chl[u].pb(v);
    }
}
vi dfsPath(int u, int target, int p) {
    if(u == target) return {u};
    FOR(v,adj[u]) if(v != p) {
        vi res = dfsPath(v,target,u);
        if(res.empty()) continue;
        res.pb(u);
        return res;
    }
    return {};
}

int dp[MX][MX];
int dpPath[N];

int getAns(int u, int bm);
int getAnsPath(int pi) {
    if(dpPath[pi] != -1) return dpPath[pi];
    int _, res;
    tie(_,_,res) = unPaved[pi];
    FOR(u,path[pi]) {
        if(u == rootPath[pi]) continue;
        res += getAns(u,childrenBitMask[u][pi]);
    }
    return dpPath[pi] = res;
}
int getAns(int u, int bm) {
    if(dp[u][bm] != -1) return dp[u][bm];

    int res = 0;
    RE(i,chl[u].size())
        if((bm&(1<<i)) == 0)
            res += getAns(chl[u][i], 0);
    
    FOR(pi,pathsOnVertex[u]) {
        if(bm&childrenBitMask[u][pi]) continue;

        int nRes = getAnsPath(pi);
        nRes += getAns(u,bm | childrenBitMask[u][pi]);
        res = max(res, nRes);
    }

    return dp[u][bm] = res;
}

void program() {
    IN(n,m);
    RE(i,m) {
        IN(a,b,c); a--; b--;
        if(c == 0) adj[a].pb(b), adj[b].pb(a);
        else unPaved.pb({a,b,c});
        totUnPaved += c;
    }
    dfsChl(0,-1);

    RE(i,unPaved.size()) {
        int u, v, d;
        tie(u,v,d) = unPaved[i];
        path[i] = dfsPath(u,v,-1);

        if((path[i].size()%2) == 0) continue;

        int highest = u;
        FOR(u,path[i]) if(depth[u] < depth[highest]) highest = u;
        pathsOnVertex[highest].pb(i);
        rootPath[i] = highest;
        FOR(u,path[i]) if(par[u] != -1) childrenBitMask[par[u]][i] |= (1<<childNumber[u]);
    }

    memset(dp,-1,sizeof(dp));
    memset(dpPath,-1,sizeof(dpPath));

    OUTL(totUnPaved - getAns(0,0));
}
# Verdict Execution time Memory Grader output
1 Correct 3 ms 4812 KB Output is correct
2 Correct 3 ms 4812 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 3 ms 5068 KB Output is correct
2 Correct 3 ms 5068 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 48 ms 20436 KB Output is correct
2 Correct 54 ms 21360 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 4 ms 4812 KB Output is correct
2 Correct 4 ms 4812 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 3 ms 4812 KB Output is correct
2 Correct 3 ms 4812 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 3 ms 4940 KB Output is correct
2 Correct 3 ms 5068 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 5 ms 5452 KB Output is correct
2 Correct 5 ms 5452 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 15 ms 6464 KB Output is correct
2 Correct 16 ms 6228 KB Output is correct
3 Correct 18 ms 8064 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 49 ms 12680 KB Output is correct
2 Correct 36 ms 7692 KB Output is correct
3 Correct 45 ms 9372 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 12 ms 5632 KB Output is correct
2 Correct 18 ms 8780 KB Output is correct
3 Correct 73 ms 23028 KB Output is correct
4 Correct 21 ms 9744 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 66 ms 14412 KB Output is correct
2 Correct 121 ms 30916 KB Output is correct
3 Correct 54 ms 12460 KB Output is correct
4 Correct 45 ms 8852 KB Output is correct