#include <stdio.h>
#include <stdlib.h>
#include <time.h>
#include <math.h>
#include <iostream>
using namespace std;

#include "lt.h"

int LT::recv(int *p)
{
  newp(p);

  int q;

  q=clean(N-1);
  if(q==0)
    remove(N-1);
  else if(q==1)
  {
    for(int i=0;i<K;i++)
      if(M[N-1][i])
      {
        node[i]=1;
        break;
      }
    remove(N-1);
    while(checkall())
      ;
  }

  q=0;
  for(int i=0;i<K;i++)
    q+=node[i];
  return q; 
  
}

int LT::checkall()
{
  for(int row=0;row<N;row++)
  {
    int q=clean(row);
    if(q==0)
    {
      remove(row);
      row--;
      continue;
    }
    else if(q==1)
    {
      for(int i=0;i<K;i++)
        if(M[row][i])
        {
          node[i]=1;
          break;
        }
      remove(row);
      return 1;
    }    
  }
  return 0;
}

void LT::remove(int row)
{
  for(int i=row;i+1<N;i++)
    for(int j=0;j<K;j++)
      M[i][j]=M[i+1][j];
  N--;
}

int LT::clean(int row)
{
  for(int i=0;i<K;i++)
    if(M[row][i] & node[i])
      M[row][i]=0;
  int q=0;
  for(int i=0;i<K;i++)
    q+=M[row][i];
  return q; 
}

void LT::init()
{
  N=0;
  for(int i=0;i<K;i++)
    node[i]=0;  
}

void LT::newp(int *p)
{
  N++;
  for(int i=0;i<K;i++)
    M[N-1][i]=p[i];
}

void LT::pr()
{
  printf("nodes:",K);
  for(int i=0;i<K;i++)
    printf(" %s",node[i]?"O":"-");
  printf("\r\n");
  //printf("M[%d][%d]=\r\n",N,K);
  for(int i=0;i<N;i++)
  {
    printf("%5d:",i);
    for(int j=0;j<K;j++)
      printf(" %s",M[i][j]?"X":"-");
    printf("\r\n");
  }
}

LT::LT(int K_, int maxN_)
{
  K=K_;
  maxN=maxN_;
  M=new (int*)[maxN];
  for(int i=0;i<maxN;i++)
    M[i]=new int[K];
  node=new int[K];
  init();
}

LT::~LT()
{
  for(int i=0;i<maxN;i++)
    delete[] M[i];
  delete[] M;
  delete[] node;
}