Submission #1350982


Source Code Expand

#include <bits/stdc++.h>
 
using namespace std;
const int INF=1e9;

int n,x;
int h[105],cnt[105]; 
vector<int> v[105];

int dfs(int x,int y){
    int ret=0;
    for (int i=0; i<v[x].size(); i++) {
        int pos=v[x][i];
        if(pos==y) continue;

        int a=dfs(pos,x);
        if(a>0||h[pos]) ret+=a+1;
    }
    return ret;
}

int main(){

cin>>n>>x; x--;

for(int i=0; i<n; i++) scanf("%d",&h[i]);

for(int i=0; i<n-1; i++){
    int a,b; scanf("%d %d",&a,&b); a--; b--;
    v[a].push_back(b);
    v[b].push_back(a);
}

v[x].push_back(-1); 

cout<<2*dfs(x,-1)<<endl;

}

Submission Info

Submission Time
Task B - ツリーグラフ
User momotaro1303
Language C++14 (GCC 5.4.1)
Score 100
Code Size 620 Byte
Status AC
Exec Time 1 ms
Memory 256 KB

Compile Error

./Main.cpp: In function ‘int main()’:
./Main.cpp:26:41: warning: ignoring return value of ‘int scanf(const char*, ...)’, declared with attribute warn_unused_result [-Wunused-result]
 for(int i=0; i<n; i++) scanf("%d",&h[i]);
                                         ^
./Main.cpp:29:34: warning: ignoring return value of ‘int scanf(const char*, ...)’, declared with attribute warn_unused_result [-Wunused-result]
     int a,b; scanf("%d %d",&a,&b); a--; b--;
                                  ^

Judge Result

Set Name Sample All
Score / Max Score 0 / 0 100 / 100
Status
AC × 2
AC × 20
Set Name Test Cases
Sample subtask0_sample_01.txt, subtask0_sample_02.txt
All subtask0_sample_01.txt, subtask0_sample_02.txt, subtask1_line01.txt, subtask1_line02.txt, subtask1_line03.txt, subtask1_line04.txt, subtask1_line05.txt, subtask1_line06.txt, subtask1_random01.txt, subtask1_random02.txt, subtask1_random03.txt, subtask1_random04.txt, subtask1_random05.txt, subtask1_random06.txt, subtask1_random07.txt, subtask1_random08.txt, subtask1_special01.txt, subtask1_special02.txt, subtask1_special03.txt, subtask1_special04.txt
Case Name Status Exec Time Memory
subtask0_sample_01.txt AC 1 ms 256 KB
subtask0_sample_02.txt AC 1 ms 256 KB
subtask1_line01.txt AC 1 ms 256 KB
subtask1_line02.txt AC 1 ms 256 KB
subtask1_line03.txt AC 1 ms 256 KB
subtask1_line04.txt AC 1 ms 256 KB
subtask1_line05.txt AC 1 ms 256 KB
subtask1_line06.txt AC 1 ms 256 KB
subtask1_random01.txt AC 1 ms 256 KB
subtask1_random02.txt AC 1 ms 256 KB
subtask1_random03.txt AC 1 ms 256 KB
subtask1_random04.txt AC 1 ms 256 KB
subtask1_random05.txt AC 1 ms 256 KB
subtask1_random06.txt AC 1 ms 256 KB
subtask1_random07.txt AC 1 ms 256 KB
subtask1_random08.txt AC 1 ms 256 KB
subtask1_special01.txt AC 1 ms 256 KB
subtask1_special02.txt AC 1 ms 256 KB
subtask1_special03.txt AC 1 ms 256 KB
subtask1_special04.txt AC 1 ms 256 KB