oj.uz

답안 #427575

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
427575 2021-06-14T17:23:00 Z davi_bart Power Plant (JOI20_power) C++17
0 / 100
 10 ms 9708 KB 
#pragma GCC optimize("O3")
#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define int ll
#define fi first
#define se second
#define ld long double
#define pb push_back
int N;
string c;
vector<int> v[200010],p[200010];
void dfs(int pos,int prec,int good){
    if(good!=-1 && c[pos]=='1'){
        p[good].pb(pos);
    }
    if(c[pos]=='1')good=pos;
    for(int x:v[pos]){
        if(x==prec)continue;
        dfs(x,pos,good);
    }
}
pair<int,int> ans(int pos,int prec){
    vector<pair<int,int> > k={{0,0}};
    for(int x:v[pos]){
        if(x==prec)continue;
        k.pb(ans(x,pos));
    }
    pair<int,int> ma={0,0};//at most 1 in line,most 2 in line
    if(c[pos]=='1'){
        for(auto [a,b]:k){
            ma.fi+=a;
            ma.se=max(ma.se,a+1);
        }
        ma.fi--;
        ma.fi=max(ma.fi,(int)1);
    }else{
        for(auto [a,b]:k){
            ma.fi+=a;
            ma.se=max(ma.se,b);
        }
    }
    //cout<<pos<<" "<<ma<<endl;
    return ma;
}
signed main(){
	ios::sync_with_stdio(false);cin.tie(0);
    cin>>N;
    for(int i=0;i<N-1;i++){
        int a,b;
        cin>>a>>b;
        a--;b--;
        v[a].pb(b);
        v[b].pb(a);
    }
    cin>>c;
    auto k=ans(0,-1);
    cout<<max(k.fi,k.se)<<endl;
}

Subtask #1
0.0 / 6.0

# 결과 실행 시간 메모리 Grader output
1 Correct 7 ms 9676 KB Output is correct
2 Correct 7 ms 9676 KB Output is correct
3 Correct 10 ms 9676 KB Output is correct
4 Correct 8 ms 9676 KB Output is correct
5 Correct 7 ms 9676 KB Output is correct
6 Correct 7 ms 9708 KB Output is correct
7 Incorrect 8 ms 9676 KB Output isn't correct
8 Halted 0 ms 0 KB -

Subtask #2
0.0 / 41.0

# 결과 실행 시간 메모리 Grader output
1 Correct 7 ms 9676 KB Output is correct
2 Correct 7 ms 9676 KB Output is correct
3 Correct 10 ms 9676 KB Output is correct
4 Correct 8 ms 9676 KB Output is correct
5 Correct 7 ms 9676 KB Output is correct
6 Correct 7 ms 9708 KB Output is correct
7 Incorrect 8 ms 9676 KB Output isn't correct
8 Halted 0 ms 0 KB -

Subtask #3
0.0 / 53.0

# 결과 실행 시간 메모리 Grader output
1 Correct 7 ms 9676 KB Output is correct
2 Correct 7 ms 9676 KB Output is correct
3 Correct 10 ms 9676 KB Output is correct
4 Correct 8 ms 9676 KB Output is correct
5 Correct 7 ms 9676 KB Output is correct
6 Correct 7 ms 9708 KB Output is correct
7 Incorrect 8 ms 9676 KB Output isn't correct
8 Halted 0 ms 0 KB -