Zoo

input/code.cpp: In function 'int main()':
input/code.cpp:63:35: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
             for (int coi = 0; coi < co[ci].size(); ++coi) {
                               ~~~~^~~~~~~~~~~~~~~

#include <bits/stdc++.h>
using namespace std;
#define F first
#define S second
int dp[10000][1<<4];
int z[50000];
int main() {
  ios::sync_with_stdio(0);
  cin.tie(0);
  int n, c;
  cin >> n >> c;
  vector<bool> co[c];
  vector<int> cow[c];
  vector<int> ctm[n][2];
  for (int i = 0; i < c; ++i) {
    int e, f, l;
    cin >> e >> f >> l;
    e--;
    for (int j = 0; j < f; ++j) {
      int x;
      cin >> x;
      x--;
      co[i].push_back(0);
      cow[i].push_back(x);
      ctm[x][0].push_back(i);
    }
    for (int j = 0; j < l; ++j) {
      int x;
      cin >> x;
      x--;
      co[i].push_back(1);
      cow[i].push_back(x);
      ctm[x][1].push_back(i);
    }
  }
  int res = 0;
  for (int sp = 0; sp < 1<<4; ++sp) {
    for (int i = 3; i < n; ++i) {
      for (int lp = 0; lp < 1<<4; ++lp) {
        dp[i][lp] = 0;
      }
    }
    for (int i = 0; i < c; ++i) z[i] = 0;
    int icc = 0;
    for (int j = 0; j < 4; ++j) {
      for (int ci : ctm[j][(sp >> j)&1]) {
        if (!z[ci]) {
          icc++;
          z[ci] = 1;
        }
      }
    }
    dp[3][sp] = icc;
    res = max(res, icc);
    for (int i = 4; i < n; ++i) {
      for (int lp = 0; lp < 1<<4; ++lp) {
        for (int pbit = 0; pbit <= 1; ++pbit) {
          int pp = ((lp << 1)&15) | pbit;
          int gc = 0;
          for (int ci : ctm[i][lp>>3]) {
            if (z[ci]) continue;
            bool f = 1;
            for (int coi = 0; coi < co[ci].size(); ++coi) {
              int idx = cow[ci][coi]-(i-4);
              int bv = co[ci][coi];
              if (idx >= 0 && idx < 4 && ((pp>>idx)&1) == bv) {
                f=0; break;
              }
            }
            gc += f;
          }
          dp[i][lp] = max(dp[i][lp], dp[i-1][pp]+gc);
          res = max(res, dp[i][lp]);
        }
      }
    }
  }
  cout << res << endl;
}

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