Submission #887559

#	Time	Username	Problem	Language	Result	Execution time	Memory
887559		amirhoseinfar1385	Cats or Dogs (JOI18_catdog)	C++17	38 / 100	3042 ms	7028 KiB

catdog

#include "catdog.h"
#include<bits/stdc++.h>
using namespace std;
const int maxn=100000+10;
vector<int>adj[maxn];
int val[maxn],n,inf=1e6+5;
void pre(int u,int par=0){
    int j=-1;
    for(int i=0;i<(int)adj[u].size();i++){
        if(adj[u][i]!=par){
            pre(adj[u][i],u);
        }
        else{
            j=i;
        }
    }
    if(j>=0){
        adj[u].erase(adj[u].begin()+j);
    }
    return ;
}
int res=0;
 
pair<int,int>cal(int u){
    pair<int,int>ret=make_pair(0,0);
    for(auto x:adj[u]){
        pair<int,int>fr=cal(x);
        ret.first+=fr.first;
        ret.second+=fr.second;
    }
    int red=0,blue=0;
    if(val[u]==1){
        res+=ret.second;
        return make_pair(1,0);
    }
    if(val[u]==2){
        res+=ret.first;
        return make_pair(0,1);
    }
    res+=min(ret.first,ret.second);
    if(ret.first==ret.second){
        return make_pair(0,0);
    }
    if(ret.first>ret.second){
        return make_pair(1,0);
    }
    return make_pair(0,1);
}
 
void initialize(int N, std::vector<int> A, std::vector<int> B) {
    n=N;
    for(int i=0;i<n-1;i++){
        adj[A[i]].push_back(B[i]);
        adj[B[i]].push_back(A[i]);
    }
    pre(1);
}
 
int cat(int v) {
    res=0;
    val[v]=1;
    cal(1);
    return res;
}
 
int dog(int v){
    res=0;
    val[v]=2;
    cal(1);
    return res;
}
 
int neighbor(int v) {
    res=0;
    val[v]=0;
    cal(1);
    return res;
}

Compilation message (stderr)

catdog.cpp: In function 'std::pair<int, int> cal(int)':
catdog.cpp:31:9: warning: unused variable 'red' [-Wunused-variable]
   31 |     int red=0,blue=0;
      |         ^~~
catdog.cpp:31:15: warning: unused variable 'blue' [-Wunused-variable]
   31 |     int red=0,blue=0;
      |               ^~~~

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