149562025-02-10 08:56:14sarminTúra (75 pont)cpp17Elfogadva 75/7554ms6452 KiB
// Created by Armin on 2/10/2025.

#include <bits/stdc++.h>
using namespace std;
using ll = long long;
using pp = pair<int, int>;
using vi = vector<int>;
using v = vector<int>;
using vvi = vector<vector<int>>;
using vpi = vector<pair<int, int>>;
using vl = vector<ll>;
constexpr ll MOD = 1e9 + 7;

void solve() {
  int n; cin >> n;
  vector<int> mi(n);
  for (int i = 0; i < n; i++) {
    cin >> mi[i];
  }
  vector<vector<int>> neighbors(n);
  for (int i = 0; i < n - 1; i++) {
    int a, b; cin >> a >> b;
    a--; b--;
    neighbors[a].push_back(b);
    neighbors[b].push_back(a);
  }

  for (int i = 0; i < n; i++) {
    vector<int> mn;
    for (int& j : neighbors[i]) {
      if (mi[j] == -1) {
        mn.push_back(j);
      }
      if (mi[i] == -1 && mi[j] == -1) {
        cout << "NEM\n2\n" << i + 1 << " " << j + 1 << "\n"; return;
      }
      if (mn.size() >= 2) {
        cout << "NEM\n3\n";
        cout << mn[0] + 1 << " " << i + 1 << " " << mn[1] + 1 << "\n"; return;
      }
    }
  }
  cout << "IGEN\n";
}

int main() {
    ios_base::sync_with_stdio(false); cin.tie(0);
    
    int t; cin >> t;
    while (t--) {
      solve();
    }
    
    return 0;
}
RészfeladatÖsszpontTesztVerdiktIdőMemória
base75/75
1Elfogadva0/01ms508 KiB
2Elfogadva0/01ms316 KiB
3Elfogadva6/61ms316 KiB
4Elfogadva3/34ms316 KiB
5Elfogadva3/34ms316 KiB
6Elfogadva4/44ms316 KiB
7Elfogadva4/44ms316 KiB
8Elfogadva4/44ms480 KiB
9Elfogadva2/239ms824 KiB
10Elfogadva2/241ms1004 KiB
11Elfogadva2/241ms828 KiB
12Elfogadva2/241ms828 KiB
13Elfogadva2/241ms828 KiB
14Elfogadva3/341ms828 KiB
15Elfogadva2/241ms832 KiB
16Elfogadva2/241ms1016 KiB
17Elfogadva2/241ms1008 KiB
18Elfogadva2/239ms976 KiB
19Elfogadva3/339ms828 KiB
20Elfogadva2/237ms824 KiB
21Elfogadva2/241ms828 KiB
22Elfogadva3/343ms828 KiB
23Elfogadva2/248ms6452 KiB
24Elfogadva2/250ms6260 KiB
25Elfogadva2/248ms3396 KiB
26Elfogadva2/243ms3828 KiB
27Elfogadva2/248ms3388 KiB
28Elfogadva2/246ms3532 KiB
29Elfogadva2/248ms3272 KiB
30Elfogadva2/245ms3608 KiB
31Elfogadva2/248ms3536 KiB
32Elfogadva2/254ms3320 KiB