提出 #73904565

ソースコード 拡げる

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

void dfs(const auto& A, const auto& graph, auto& visited, auto& result, int parent, int u, int flag) {
	if (visited[A[u]]) {
		flag = 1;
	}
	result[u] = flag;
	++visited[A[u]];
	for (int v : graph[u]) {
		if (v != parent) {
			dfs(A, graph, visited, result, u, v, flag);
		}
	}
	--visited[A[u]];
}

signed main() {
	ios::sync_with_stdio(0);
	cin.tie(nullptr);

	int N;
	cin >> N;

	vector<int> A(N + 1);
	for (int i = 1; i <= N; ++i) {
		cin >> A[i];
	}
	vector<vector<int>> graph(N + 1);
	for (int i = 0; i < N - 1; ++i) {
		int u, v;
		cin >> u >> v;
		graph[u].push_back(v);
		graph[v].push_back(u);
	}
	map<int, int> visited;
	vector<int> result(N + 1);
	dfs(A, graph, visited, result, -1, 1, 0);
	for (int i = 1; i <= N; ++i) {
		if (result[i]) {
			cout << "Yes\n";
		}
		else {
			cout << "No\n";
		}
	}
}

提出情報

提出日時
問題 D - Integer-duplicated Path
ユーザ bluerini
言語 C++23 (GCC 15.2.0)
得点 400
コード長 928 Byte
結果 AC
実行時間 247 ms
メモリ 54996 KiB

ジャッジ結果

セット名 Sample All
得点 / 配点 0 / 0 400 / 400
結果
AC × 3
AC × 47
セット名 テストケース
Sample sample_01.txt, sample_02.txt, sample_03.txt
All hack_01.txt, hack_02.txt, hack_03.txt, hack_04.txt, hack_05.txt, hack_06.txt, sample_01.txt, sample_02.txt, sample_03.txt, test_01.txt, test_02.txt, test_03.txt, test_04.txt, test_05.txt, test_06.txt, test_07.txt, test_08.txt, test_09.txt, test_10.txt, test_11.txt, test_12.txt, test_13.txt, test_14.txt, test_15.txt, test_16.txt, test_17.txt, test_18.txt, test_19.txt, test_20.txt, test_21.txt, test_22.txt, test_23.txt, test_24.txt, test_25.txt, test_26.txt, test_27.txt, test_28.txt, test_29.txt, test_30.txt, test_31.txt, test_32.txt, test_33.txt, test_34.txt, test_35.txt, test_36.txt, test_37.txt, test_38.txt
ケース名 結果 実行時間 メモリ
hack_01.txt AC 176 ms 36996 KiB
hack_02.txt AC 178 ms 36988 KiB
hack_03.txt AC 42 ms 19120 KiB
hack_04.txt AC 42 ms 19068 KiB
hack_05.txt AC 121 ms 33796 KiB
hack_06.txt AC 122 ms 33788 KiB
sample_01.txt AC 1 ms 3548 KiB
sample_02.txt AC 1 ms 3580 KiB
sample_03.txt AC 1 ms 3644 KiB
test_01.txt AC 1 ms 3604 KiB
test_02.txt AC 1 ms 3464 KiB
test_03.txt AC 245 ms 54996 KiB
test_04.txt AC 247 ms 54852 KiB
test_05.txt AC 245 ms 54920 KiB
test_06.txt AC 216 ms 49864 KiB
test_07.txt AC 84 ms 42580 KiB
test_08.txt AC 80 ms 42368 KiB
test_09.txt AC 154 ms 31644 KiB
test_10.txt AC 149 ms 31476 KiB
test_11.txt AC 148 ms 31468 KiB
test_12.txt AC 120 ms 23224 KiB
test_13.txt AC 46 ms 18916 KiB
test_14.txt AC 41 ms 19068 KiB
test_15.txt AC 186 ms 32276 KiB
test_16.txt AC 191 ms 32292 KiB
test_17.txt AC 188 ms 32328 KiB
test_18.txt AC 187 ms 30768 KiB
test_19.txt AC 68 ms 19748 KiB
test_20.txt AC 62 ms 19796 KiB
test_21.txt AC 193 ms 33112 KiB
test_22.txt AC 164 ms 31572 KiB
test_23.txt AC 179 ms 32000 KiB
test_24.txt AC 125 ms 21980 KiB
test_25.txt AC 60 ms 19740 KiB
test_26.txt AC 55 ms 20252 KiB
test_27.txt AC 234 ms 49152 KiB
test_28.txt AC 210 ms 38744 KiB
test_29.txt AC 241 ms 48804 KiB
test_30.txt AC 203 ms 42072 KiB
test_31.txt AC 75 ms 31888 KiB
test_32.txt AC 75 ms 34684 KiB
test_33.txt AC 205 ms 40720 KiB
test_34.txt AC 237 ms 49624 KiB
test_35.txt AC 226 ms 45020 KiB
test_36.txt AC 168 ms 27468 KiB
test_37.txt AC 84 ms 39256 KiB
test_38.txt AC 67 ms 24012 KiB