Submission #1070898


Source Code Expand

#include<bits/stdc++.h>
#define rep(i,n)for(int i=0;i<(n);i++)
using namespace std;

int h[100], d[100], in[100], c[100];
vector<int>E[100], G[100], v;

int main() {
	int n, x; scanf("%d%d", &n, &x); x--;
	rep(i, n)scanf("%d", &h[i]);
	rep(i, n - 1) {
		int a, b; scanf("%d%d", &a, &b); a--; b--;
		E[a].push_back(b); E[b].push_back(a);
	}
	memset(d, -1, sizeof(d)); d[x] = 0;
	queue<int>Q; Q.push(x);
	while (!Q.empty()) {
		int p = Q.front(); Q.pop();
		for (int u : E[p]) {
			if (d[u] == -1) {
				d[u] = d[p] + 1; Q.push(u);
				G[p].push_back(u); in[u]++;
			}
		}
	}
	Q.push(x);
	while (!Q.empty()) {
		int p = Q.front(); Q.pop();
		v.push_back(p);
		for (int u : G[p]) {
			if (--in[u] == 0)Q.push(u);
		}
	}
	reverse(v.begin(), v.end());
	for (int i : v) {
		for (int u : G[i]) {
			if (c[u] || h[u])c[i] += c[u] + 1;
		}
	}
	printf("%d\n", c[0] * 2);
}

Submission Info

Submission Time
Task B - ツリーグラフ
User autumn_eel
Language C++11 (GCC 4.8.1)
Score 0
Code Size 902 Byte
Status WA
Exec Time 20 ms
Memory 924 KB

Compile Error

./Main.cpp: In function ‘int main()’:
./Main.cpp:9:33: warning: ignoring return value of ‘int scanf(const char*, ...)’, declared with attribute warn_unused_result [-Wunused-result]
  int n, x; scanf("%d%d", &n, &x); x--;
                                 ^
./Main.cpp:10:29: warning: ignoring return value of ‘int scanf(const char*, ...)’, declared with attribute warn_unused_result [-Wunused-result]
  rep(i, n)scanf("%d", &h[i]);
                             ^
./Main.cpp:12: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 0 / 100
Status
AC × 2
AC × 12
WA × 8
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 20 ms 912 KB
subtask0_sample_02.txt AC 19 ms 916 KB
subtask1_line01.txt AC 19 ms 912 KB
subtask1_line02.txt AC 17 ms 840 KB
subtask1_line03.txt AC 20 ms 920 KB
subtask1_line04.txt WA 19 ms 916 KB
subtask1_line05.txt WA 18 ms 916 KB
subtask1_line06.txt AC 19 ms 920 KB
subtask1_random01.txt AC 19 ms 924 KB
subtask1_random02.txt AC 19 ms 920 KB
subtask1_random03.txt WA 19 ms 920 KB
subtask1_random04.txt WA 19 ms 920 KB
subtask1_random05.txt WA 17 ms 920 KB
subtask1_random06.txt WA 19 ms 920 KB
subtask1_random07.txt WA 17 ms 916 KB
subtask1_random08.txt WA 17 ms 920 KB
subtask1_special01.txt AC 19 ms 920 KB
subtask1_special02.txt AC 19 ms 920 KB
subtask1_special03.txt AC 19 ms 916 KB
subtask1_special04.txt AC 19 ms 916 KB