152662025-02-17 18:55:02TakacsAndrasTúra (75 pont)cpp17Wrong answer 4/75118ms6964 KiB
#include <bits/stdc++.h>
using namespace std;
vector<int> a,p;
vector<vector<int>> g;
bool e = false;
void dfs(int u, int pr, int cnt=0) {
    a[u] = -2;
    p[u] = pr;
    if (cnt < 0 && u != 1) {
        int h = u;
        vector<int> out;
        while (h != -1) {
            out.push_back(h);
            h = p[h];
        }
        cout << "NEM\n" << out.size() << '\n';
        for (int x : out) cout << x << ' ';
        cout << '\n';
        e = true;

    }
    for (int &x : g[u]) {
        if (a[x] != -2) dfs(x,u,cnt+a[x]);
        if (e) break;
    }
}
int main() {
    int t; cin >> t;
    while (t--) {
        int n; cin >> n;
        e = false;
        a.assign(n+1,0);
        p.assign(n+1,0);
        for (int i = 1; i <= n; i++) cin >> a[i];
        g.assign(n+1,vector<int>(0));
        for (int i = 0; i < n-1; i++) {
            int a,b; cin >> a >> b;
            g[a].push_back(b);
            g[b].push_back(a);
        }
        dfs(1,-1,a[1]);
        if (!e) cout << "IGEN\n";
    }
}
SubtaskSumTestVerdictTimeMemory
base4/75
1Accepted0/01ms316 KiB
2Wrong answer0/01ms508 KiB
3Wrong answer0/61ms316 KiB
4Wrong answer0/38ms316 KiB
5Wrong answer0/38ms484 KiB
6Wrong answer0/48ms456 KiB
7Wrong answer0/48ms480 KiB
8Wrong answer0/48ms476 KiB
9Wrong answer0/283ms1128 KiB
10Wrong answer0/283ms1176 KiB
11Wrong answer0/286ms1064 KiB
12Wrong answer0/285ms1124 KiB
13Wrong answer0/285ms1072 KiB
14Wrong answer0/385ms1076 KiB
15Wrong answer0/283ms1080 KiB
16Wrong answer0/285ms1052 KiB
17Wrong answer0/285ms980 KiB
18Wrong answer0/282ms1328 KiB
19Wrong answer0/382ms1332 KiB
20Wrong answer0/282ms1332 KiB
21Wrong answer0/283ms1332 KiB
22Wrong answer0/386ms1076 KiB
23Wrong answer0/2108ms6964 KiB
24Wrong answer0/2118ms6708 KiB
25Wrong answer0/2104ms3636 KiB
26Wrong answer0/2101ms4012 KiB
27Wrong answer0/2104ms3636 KiB
28Wrong answer0/2101ms3892 KiB
29Wrong answer0/2109ms3572 KiB
30Wrong answer0/2101ms3888 KiB
31Accepted2/2101ms3892 KiB
32Accepted2/2105ms3380 KiB