#include "coreputer.h"

#include <iostream>

std::vector<int> malfunctioning_cores(int N) {
	std::vector<int> T;
	std::vector<int> c(N);
	std::vector<int> results(N);
	for(int i=0; i<N; i++) c[i] = 0;

	int a = 0, b = N-1;
	int finish = 0;
	while(a!=b){
		int half = (a+b)/2;
		T.clear();
		for(int i=0; i<=half; i++) T.push_back(i);

		results[half] = run_diagnostic(T);
		if(results[half]>=0){
			b = half;
		} else {
			a = half+1;
		}
	}

	finish = results[a];

	int h = a;
	c[h] = 1;

	if(h!=N-1){
		for(int i=0; i<h; i++){
			T.clear();
			for(int j=0; j<=h; j++){
				if(i!=j) T.push_back(j);
			}

			if(run_diagnostic(T)==-1) c[i] = 1;
		}

		h--;
		for(int i=h+2; i<N-1; i++){
			T.clear();
			for(int j=h+1; j<N; j++){
				if(i!=j) T.push_back(j);
			}

			if(run_diagnostic(T)<=0) c[i] = 1;
		}

		int fp = 0, sp = 0;
		for(int i=0; i<=h+1; i++) fp += c[i];
		for(int i=h+2; i<N; i++) sp += c[i];

		if((finish == 0 && fp > sp) || (finish == 1 && fp-1 > sp)) c[N-1] = 1;
	}

	return c;
}