Submission #978778

#TimeUsernameProblemLanguageResultExecution timeMemory
978778AdamGSSplit the Attractions (IOI19_split)C++17
40 / 100
88 ms29672 KiB
#include "split.h"
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
#define rep(a, b) for(int a = 0; a < (b); ++a)
#define st first
#define nd second
#define pb push_back
#define all(a) a.begin(), a.end()
const int LIM=1e5+7;
vector<int>V[LIM], P[LIM], G[LIM], ans;
vector<pair<int,int>>S;
int pre[LIM], odl[LIM], ile[LIM], lpre, X, Y;
void DFS(int x) {
  pre[x]=++lpre;
  for(auto i : V[x]) {
    if(!pre[i]) {
      P[x].pb(i);
      DFS(i);
    } else if(pre[i]<pre[x]) G[x].pb(i);
  }
}
void DFS2(int x) {
  ile[x]=1;
  for(auto i : P[x]) {
    odl[i]=odl[x]+1;
    DFS2(i);
    ile[x]+=ile[i];
  }
}
void DFS3(int x, int k) {
  if(!X || ans[x]) return;
  ans[x]=k;
  --X;
  for(auto i : P[x]) DFS3(i, k);
}
void DFS4(int x, int k) {
  if(ans[x]) return;
  ans[x]=k;
  for(auto i : P[x]) DFS4(i, k);
}
void DFS7(int x, int k) {
  if(X) ans[x]=k; else ans[x]=S[2].nd;
  for(auto i : P[x]) DFS7(i, k);
}
void DFS5(int x, bool c, int k, int p) {
  for(auto i : G[x]) if(odl[i]<k) c=true;
  if(c) {
    if(X) DFS7(x, p);
    return;
  }
  for(auto i : P[x]) DFS5(i, c, k, p);
}
bool DFS6(int x) {
  bool c=true;
  if(S[2].st==0) c=false;
  if(ans[x]==S[1].nd) c=false;
  for(auto i : P[x]) c&=DFS6(i);
  if(c) {
    --S[2].st;
    ans[x]=S[2].nd;
  }
  return c;
}
vector<int>find_split(int n, int a, int b, int c, vector<int>p, vector<int>q) {
  rep(i, n) ans.pb(0);
  S.pb({a, 1});
  S.pb({b, 2});
  S.pb({c, 3});
  sort(all(S));
  rep(i, p.size()) {
    V[p[i]].pb(q[i]);
    V[q[i]].pb(p[i]);
  }
  DFS(0);
  DFS2(0);
  rep(i, n) if(ile[i]>=S[0].st && n-ile[i]>=S[1].st) {
    X=S[0].st;
    DFS3(i, S[0].nd);
    X=S[1].st;
    DFS3(0, S[1].nd);
    rep(j, n) if(!ans[j]) ans[j]=S[2].nd;
    return ans;
  }
  rep(i, n) if(ile[i]>=S[1].st && n-ile[i]>=S[0].st) {
    X=S[1].st;
    DFS3(i, S[1].nd);
    X=S[0].st;
    DFS3(0, S[0].nd);
    rep(j, n) if(!ans[j]) ans[j]=S[2].nd;
    return ans;
  }
  pair<int,int>mi={n, n};
  rep(i, n) if(ile[i]>=S[0].st) mi=min(mi, {ile[i], i});
  if(!mi.nd) return ans;
  X=S[1].st-(n-mi.st);
  DFS4(mi.nd, S[0].nd);
  DFS4(0, S[1].nd);
  for(auto i : P[mi.nd]) DFS5(i, false, odl[mi.nd], S[1].nd);
  if(!X) {
    DFS6(0);
    return ans;
  }
  rep(i, n) ans[i]=0;
  return ans;
}

Compilation message (stderr)

split.cpp: In function 'std::vector<int> find_split(int, int, int, int, std::vector<int>, std::vector<int>)':
split.cpp:5:36: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
    5 | #define rep(a, b) for(int a = 0; a < (b); ++a)
      |                                    ^
split.cpp:71:3: note: in expansion of macro 'rep'
   71 |   rep(i, p.size()) {
      |   ^~~
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...