Submission #94902

#	Time	Username	Problem	Language	Result	Execution time	Memory
94902		WLZ	Untitled (POI11_kon)	C++17	90 / 100	3071 ms	102776 KiB

Main

#include <bits/stdc++.h>
using namespace std;
 
int n;
vector< vector<short> > g, tg;
vector< bitset<5001> > mat;
vector<int> was, topoSort, scc;
int cur = 1;
vector<int> as;
vector<int> st; 

void reverse_graph() {
  for (int u = 1; u < 2 * n + 1; u ++) {
    for (auto& v: g[u]) {
      tg[v].push_back(u);
    }
    vector<short>().swap(g[u]);
  }
}

void add_edge(int u, int v) {
  g[u].push_back(v);
}
 
void dfs1(int u) {
  was[u] = 1;
  for (auto& v : g[u]) {
    if (!was[v]) {
      dfs1(v);
    }
  }
  topoSort.push_back(u);
}
 
void dfs2(int u) {
  scc[u] = cur;
  was[u] = 1;
  for (auto& v : tg[u]) {
    if (!was[v]) {
      dfs2(v);
    }
  }
}
 
int main() {
  ios::sync_with_stdio(false);
  cin.tie(0);
  cin >> n;
  mat.assign(n + 1, bitset<5001>());
  g.resize(2 * n + 1);
  tg.resize(2 * n + 1);
  for (int i = 1; i <= n; i++) {
    int k;
    cin >> k;
    while (k--) {
      int a;
      cin >> a;
      mat[a][i] = mat[i][a] = 1;
    }
  }
  for (int i = 1; i <= n - 1; i++) {
    for (int j = i + 1; j <= n; j++) {
      if (mat[i][j]) {
        add_edge(i, j + n);
        add_edge(j, i + n);
      } else {
        add_edge(i + n, j);
        add_edge(j + n, i);
      }
    }
  }
  was.assign(2 * n + 1, 0);
  for (int i = 1; i <= 2 * n; i++) {
    if (!was[i]) {
      dfs1(i);
    }
  }
  reverse(topoSort.begin(), topoSort.end());
  reverse_graph();
  scc.assign(2 * n + 1, -1);
  was.assign(2 * n + 1, 0);
  for (auto& u : topoSort) {
    if (!was[u]) {
      dfs2(u);
      cur++;
    }
  }
  as.resize(n + 1);
  for (int i = 1; i <= n; i++) {
    if (scc[i] == scc[i + n]) {
      cout << 0 << '\n';
      return 0;
    }
    as[i] = scc[i] > scc[i + n];
  }
  int ans = 1;
  int sup = 0, con = 0;
  for (int i = 1; i <= n; i++) {
    sup += 1 - as[i];
    con += as[i];
  }
  if ((sup == 0) || (con == 0)) {
    ans = 0;
  } 
  for (int i = 1; i <= n; i++) {
    if (as[i]) {
      int pos = 1;
      if (con == 1) {
        pos = 0;
      }
      for (int j = 1; j <= n; j++) {
        if (!as[j] && !mat[i][j]) {
          pos = 0;
          break;
        }
      }
      ans += pos;
    } else {
      int pos = 1;
      if (sup == 1) {
        pos = 0;
      }
      for (int j = 1; j <= n; j++) {
        if (as[j] && mat[i][j]) {
          pos = 0;
          break;
        }
      }
      ans += pos;
    }
  }
  st.resize(n + 1);
  for (int i = 1; i <= n; i++) {
    int cnt = 0;
    for (int j = 1; j <= n; j++) {
      if (j == i) {
        continue;
      }
      if (as[i] && mat[i][j]) {
        cnt++;
      } 
      if (!as[i] && !mat[i][j]) {
        cnt++;
      }
    }
    for (int j = 1; j <= n; j++) {
      if (as[i] && !as[j] && !mat[i][j]) {
        st[i] = j;
      }
      if (!as[i] && as[j] && mat[i][j]) {
        st[i] = j;
      }
    }
    if (as[i] && cnt < sup - 1) {
      st[i] = -1;
    } 
    if (!as[i] && cnt < con - 1) {
      st[i] = -1;
    }
  }
  for (int i = 1; i <= n - 1; i++) {
    for (int j = i + 1; j <= n; j++) {
      if (as[i] != as[j]) {
        int x = i, y = j;
        if (as[x]) {
          swap(x, y);
        }
        if ((st[x] == 0 || st[x] == y) && (st[y] == 0 || st[y] == x)) {
          ans++;
        }
      }
    }
  }
  cout << ans << '\n';
  return 0;
}

• Subtask 1: 10 / 10
• Subtask 2: 10 / 10
• Subtask 3: 10 / 10
• Subtask 4: 10 / 10
• Subtask 5: 10 / 10
• Subtask 6: 10 / 10
• Subtask 7: 10 / 10
• Subtask 8: 10 / 10
• Subtask 9: 10 / 10
• Subtask 10: 0 / 10