Submission #796521

#	Time	Username	Problem	Language	Result	Execution time	Memory
796521		vjudge1	Split the Attractions (IOI19_split)	C++17	0 / 100	777 ms	1048576 KiB

split

#include "split.h"
#include "bits/stdc++.h"
using namespace std;

#define vi vector<int> 

const int mxn=1e5+5;
vector<int> adj[mxn];

void dfs(int u, int pa, vi &ord){
  ord.push_back(u);
  for(int v: adj[u])
    if(v!=pa) dfs(v,u,ord);
}

vi find_split(int n, int a, int b, int c, vi p, vi q) {
  int m=(int)p.size();
  for(int i=0;i<m;i++)
    adj[p[i]].push_back(q[i]),adj[q[i]].push_back(p[i]);

  int u=0;
  for(int i=0;i<n;i++)
    if(adj[i].size()==1) u=i;

  vi ord;
  dfs(u,-1,ord);

  vector<int> res(n);
  for(int i=0;i<n;i++){
    for(;i<a;i++) res[ord[i]]=1;
    for(;i<a+b;i++) res[ord[i]]=2;
    for(;i<a+b+c;i++) res[ord[i]]=3;
  }

  return res;
}

• Subtask 1: 0 / 7
• Subtask 2: 0 / 11
• Subtask 3: 0 / 22
• Subtask 4: 0 / 24
• Subtask 5: 0 / 36