#include "supertrees.h" #include <vector> using namespace std; int construct(vector< vector<int> > p) { int n = p.size(); vector< vector<int> > b (n, vector<int>(n, 0)); vector<int> oncycle(n, 1); for(int i = 0; i < n; i++) { for(int j = i-1; j >= 0; j--) if(p[i][j] == 1) { oncycle[i] = 0; b[i][j] = b[j][i] = 1; break; } } int pre; for(int i = 0; i < n; i++) { if(!oncycle[i]) continue; pre = i; oncycle[i] = 0; for(int j = i+1; j < n; j++) { if(!oncycle[j]) continue; if(p[pre][j] == 2) { b[pre][j] = b[j][pre] = 1; pre = j; oncycle[j] = 0; } } if(pre != i) b[pre][i] = b[i][pre] = 1; } build(b); return 1; }