제출 #530309

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
530309		kevin	Power Plant (JOI20_power)	C++17	0 / 100	3 ms	5016 KiB

power

#include <bits/stdc++.h>
using namespace std;

#define ll long long
#define nl cout<<"\n"
#define all(x) x.begin(), x.end()
#define f first
#define s second
#define ca(v) for(auto i:v) cout<<i<<" ";

const int MOD = 1e9 + 7;
const int MAXN = 2e5 + 5;

vector<int> adj[MAXN];
int ar[MAXN];

int mx = 0;

int dfs(int x, int p){
    int sm = 0;
    for(int i:adj[x]){
        if(i == p) continue;
        int val = dfs(i, x);
        if(val > 0) sm += val;
        if(ar[x]) mx = max(mx, 1 + val);
    }
    if(ar[x]) {
        // cout<<x+1<<" "<<max(sm-1, 1)<<"\n";
        mx = max(mx, sm - 1);
        return max(sm-1, 1);
    }
    // cout<<x+1<<" "<<sm<<"\n";
    mx = max(mx, sm);
    return sm;
}

int main()
{
    ios_base::sync_with_stdio(0); cin.tie(0);
    if (fopen("input.in", "r")) freopen("input.in", "r", stdin);
    int n;
    cin>>n;
    for(int i=0; i<n-1; i++){
        int u, v; cin>>u>>v;
        adj[--u].push_back(--v);
        adj[v].push_back(u);
    }
    for(int i=0; i<n; i++){
        char c; cin>>c;
        ar[i] = c == '1';
    }
    dfs(0, -1);
    cout<<mx;
}

컴파일 시 표준 에러 (stderr) 메시지

power.cpp: In function 'int main()':
power.cpp:40:40: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   40 |     if (fopen("input.in", "r")) freopen("input.in", "r", stdin);
      |                                 ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~

• Subtask 1: 0 / 6
• Subtask 2: 0 / 41
• Subtask 3: 0 / 53