Submission #388117

#	Time	Username	Problem	Language	Result	Execution time	Memory
388117		mohamedsobhi777	Cats or Dogs (JOI18_catdog)	C++14	8 / 100	10 ms	520 KiB

catdog

#include <bits/stdc++.h>
#include "catdog.h"

using namespace std;

int x;
const int MX = 1000 + 7;
int val[MX];
vector<int> adj[MX];
int root;
int dp[MX][2] ;

void initialize(int N, std::vector<int> A, std::vector<int> B)
{
       x = N ; 
       for (int i = 0; i < N; ++i)
       {
              adj[A[i]].push_back(B[i]);
              adj[B[i]].push_back(A[i]);
       }
       fill(val + 1, val + N + 1, 2) ;
}

void dfs(int x, int p)
{
       if(val[x]!=2)dp[x][ !val[x] ] = MX ;
       for(auto u : adj[x]){
              if(u == p)continue; 
              dfs(u , x) ;
              dp[x][0] += min(dp[u][0] ,dp[u][1] + 1) ; 
              dp[x][1] += min(dp[u][1] ,dp[u][0] + 1) ;
       }
}

int solve(){
       memset(dp ,0 ,sizeof dp) ;
       dfs(1 , 1) ; 
       return min(dp[1][0] , dp[1][1]) ;
}

int cat(int v)
{
       val[v] = 0;
       return solve() ; 
}

int dog(int v)
{
       val[v] = 1;
       return solve() ;
}

int neighbor(int v)
{
       val[v] = 2;
       return solve() ;
}

• Subtask 1: 8 / 8
• Subtask 2: 0 / 30
• Subtask 3: 0 / 62