Online feud

#include <bits/stdc++.h>
using namespace std;
 
#define PRll_ARR(arr, n)                                                      \
  do {                                                                         \
    for (ll i = 0; i < n; i++) {                                              \
      cout << arr[i] << " ";                                                   \
    }                                                                          \
    cout << "\n";                                                              \
  } while (0)
#define PRll_VEC_ARR(v, n)                                                    \
  do {                                                                         \
    for (ll i = 0; i < n; i++) {                                              \
      cout << i << ": ";                                                       \
      for (auto item : v[i]) {                                                 \
        cout << item.first << " " ;                                                   \
      }                                                                        \
      cout << endl;                                                            \
    }                                                                          \
  } while (0)
 
 
typedef long long ll;
typedef pair<ll, bool> pp;
 
bool dfs(ll i, ll c, ll *color, vector<pp> *status, ll n) {
  bool s = true;
  for (auto p : status[i]) {
    if (p.second) {
      if (color[p.first] != 0 && color[p.first] != c) {
        return false;
      } else if (color[p.first] == 0) {
        color[p.first] = c;
        s = dfs(p.first, c, color, status, n);
      } else {
        continue;
      }
    } else {
      if (color[p.first] == c) {
        return false;
      } else if (color[p.first] == 0){
        color[p.first] = c == 1 ? -1 : 1;
        s = dfs(p.first, (c == 1 ? -1 : 1), color, status, n);
      } else {
        continue;
      }
    }
    if (!s)
      return false;
  }
  return true;
}
 
int main() {
  ios::sync_with_stdio(0);
  cin.tie(0);
 
  ll n, m;
  cin >> n >> m;
  ll arr[n];
  memset(arr, 0, n * sizeof(ll));
  arr[0] = 1;
  vector<pp> status[n];
 
  ll a, b, c;
  for (ll i = 0; i < m; i++) {
    cin >> a >> b >> c;
    a--;
    b--;
    status[a].push_back({b, (c == 1)});
    status[b].push_back({a, (c == 1)});
  }
 
  ll color[n];
  memset(color, 0, n*sizeof(ll));
  color[0] = 1;
  bool state = dfs(0, 1, color, status, n);
 
 
  cout << (state ? "Yes" : "No");
  return 0;
}

Test 1

Verdict: ACCEPTED

Test 2

Verdict: ACCEPTED

Test 3

Verdict: ACCEPTED

Test 4

Verdict: ACCEPTED

Test 5

Verdict: ACCEPTED

Test 6

Verdict: ACCEPTED

Test 7

Verdict: ACCEPTED

Test 8

Verdict: ACCEPTED

Test 9

Verdict: ACCEPTED

Test 10

Verdict: ACCEPTED

Test 11

Verdict: ACCEPTED

Test 12

Verdict: ACCEPTED

Test 13

Verdict: ACCEPTED

Test 14

Verdict: ACCEPTED

Test 15

Verdict: ACCEPTED

Test 16

Verdict: ACCEPTED

Test 17

Verdict: ACCEPTED

Test 18

Verdict: ACCEPTED

Test 19

Verdict: ACCEPTED

Test 20

Verdict: ACCEPTED

Test 21

Verdict: ACCEPTED

Test 22

Verdict: ACCEPTED

Test 23

Verdict: WRONG ANSWER

Test 24

Verdict: ACCEPTED

Test 25

Verdict: ACCEPTED

Test 26

Verdict: ACCEPTED

Test 27

Verdict: ACCEPTED

Test 28

Verdict: ACCEPTED

Test 29

Verdict: ACCEPTED

Test 30

Verdict: ACCEPTED

Test 31

Verdict: ACCEPTED

Test 32

Verdict: ACCEPTED

Test 33

Verdict: ACCEPTED

Test 34

Verdict: ACCEPTED

Test 35

Verdict: ACCEPTED

Test 36

Verdict: ACCEPTED

Test 37

Verdict: ACCEPTED

Test 38

Verdict: ACCEPTED

Test 39

Verdict: ACCEPTED

Test 40

Verdict: ACCEPTED

Test 41

Verdict: ACCEPTED

Test 42

Verdict: ACCEPTED

Test 43

Verdict: WRONG ANSWER

Test 44

Verdict: ACCEPTED

Test 45

Verdict: ACCEPTED

Test 46

Verdict: ACCEPTED

Test 47

Verdict: ACCEPTED

Test 48

Verdict: ACCEPTED

Test 49

Verdict: ACCEPTED

Test 50

Verdict: ACCEPTED

Test 51

Verdict: ACCEPTED

Test 52

Verdict: ACCEPTED

Test 53

Verdict: ACCEPTED

Test 54

Verdict: ACCEPTED

Test 55

Verdict: ACCEPTED

Test 56

Verdict: ACCEPTED

Test 57

Verdict: ACCEPTED

Test 58

Verdict: ACCEPTED

Test 59

Verdict: ACCEPTED

Test 60

Verdict: ACCEPTED

Test 61

Verdict: ACCEPTED

Test 62

Verdict: ACCEPTED

Test 63

Verdict: WRONG ANSWER

Test 64

Verdict: ACCEPTED

Test 65

Verdict: ACCEPTED

Test 66

Verdict: ACCEPTED

Test 67

Verdict: ACCEPTED

Test 68

Verdict: ACCEPTED

Test 69

Verdict: ACCEPTED

Test 70

Verdict: ACCEPTED

Test 71

Verdict: ACCEPTED