Submission #283972

#TimeUsernameProblemLanguageResultExecution timeMemory
283972milleniumEeeeMechanical Doll (IOI18_doll)C++14
0 / 100
32 ms4300 KiB
#include "doll.h"
#include <bits/stdc++.h>
#define fr first
#define sc second
#define pii pair<int, int>
using namespace std;
 
int N, M, S;
vector <int> A, C, X, Y;
 
struct Solve_for_permutaion {
  bool need() {
    vector <int> cnt((int)4e5 + 5, 0);
    bool ok = 1;
    for (int i = 1; i <= N; i++) {
      cnt[A[i]]++;
    }
    for (int i = 1; i <= M; i++) {
      ok &= (cnt[A[i]] <= 1);
    }
    return ok;
  }
  void solve() {
    for (int i = 1; i <= N; i++) {
      C[A[i - 1]] = A[i];
    }
    C[A[N]] = 0;
  }
}subtask1;
 
struct Solve_for_two {
  vector <int> cnt;
  Solve_for_two() {
    cnt.resize((int)1e5 + 5, 0);
  }
  bool need() {
    bool ok = 1;
    for (int i = 1; i <= N; i++) {
      cnt[A[i]]++;
    }
    for (int i = 1; i <= M; i++) {
      ok &= (cnt[A[i]] <= 2);
    }
    return ok;
  }
  
  struct Para {
    int vertex, trigger;
    Para (int vertex_, int trigger_ ) {
      vertex = vertex_, trigger = trigger_;      
    }
    Para () {
      vertex = -1, trigger = -1;
    }
  };
  
  void solve() {
    vector <bool> flag(M + 5, 0);
    vector <int> pos(M + 5, 0);
    vector <pii> twice(M + 5, {-1, -1});
    for (int i = 1; i <= N; i++) {
      pos[A[i]] = i;
      if (!pos[A[i]]) {
        twice[A[i]].fr = i;
      } else {
        twice[A[i]].sc = i;
      }
    }
    for (int i = 1; i <= M; i++) {
      flag[i] = (cnt[i] == 2);
    }
    int id = -1;
    int siz = 0;
    C[0] = A[1];
    
    vector <Para> vec;
    
    for (int i = 1; i <= M; i++) {
      if (cnt[i] == 0) {
        continue;
      }
      if (cnt[i] == 1) {
        int nxt = (pos[A[i]] < N ? A[pos[A[i]] + 1] : 0);
        C[A[i]] = nxt;
      }
      if (cnt[i] == 2) {
        C[A[i]] = id--, siz++;
        vec.push_back(Para(i, C[A[i]]));// our vertex and his trigger
      }
    }
    X.resize(siz);
    Y.resize(siz);
    for (auto el : vec) {
      int id = (-el.trigger) - 1;
      X[id] = A[twice[el.vertex].fr + 1];
      if (twice[el.vertex].sc == N) {
        Y[id] = 0;
      } else {
        Y[id] = A[twice[el.vertex].sc + 1];
      }
    }
  }
};
 
// A - sequance 
// C - edges of triggers
// X/Y - edges of switches
// M - triggers vertexes
 
void create_circuit(int MM, vector<int> AA) {
  M = MM;
  N = AA.size();
  A.resize(N + 1);
  A[0] = 0;
  for (int i = 0; i < N; i++) {
    A[i + 1] = AA[i];
  }
  C.resize(M + 1);
  if (subtask1.need()) {
    subtask1.solve();
  } else {
    assert(false);
  }
  answer(C, X, Y);
}
#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...
#Verdict Execution timeMemoryGrader output
Fetching results...