nodearray.cpp

来自「Dijkstra算法的实现」· C++ 代码 · 共 81 行

// NodeArray.cpp: implementation of the CNodeArray class.
//
//////////////////////////////////////////////////////////////////////

#include "stdafx.h"
#include "Dijkstra.h"
#include "NodeArray.h"

#ifdef _DEBUG
#undef THIS_FILE
static char THIS_FILE[]=__FILE__;
#define new DEBUG_NEW
#endif

//////////////////////////////////////////////////////////////////////
// Construction/Destruction
//////////////////////////////////////////////////////////////////////

CNodeArray::CNodeArray()
{

}

CNodeArray::~CNodeArray()
{

}

BOOL CNodeArray::AddNode(LPCTSTR strip)
{
	char *pdest = strrchr(strip,'.');
	UINT num = atoi(pdest+1);

	int count = m_ArrayNodeInfo.GetSize();
	for(int i = 0; i<count;i++)
	{
		if(m_ArrayNodeInfo[i].NodeNum == num)
			return FALSE;
	}
	NodeInfo node;
	node.NodeNum = num;
	strcpy(node.ip,strip);
	node.Weight = -1;
	m_ArrayNodeInfo.Add(node);

	return TRUE;
}



BOOL CNodeArray::DelNode(UINT num)
{
	int count = m_ArrayNodeInfo.GetSize();
	for (int i=0;i<count;i++)
	{
		if(m_ArrayNodeInfo[i].NodeNum == num)
		{
			m_ArrayNodeInfo.RemoveAt(i);
			return TRUE;
		}
	}
	return FALSE;
}

BOOL CNodeArray::DelNode(LPCTSTR lpctip)
{
	char *pdest = strrchr(lpctip,'.');
	UINT num = atoi(pdest+1);
	
	int count = m_ArrayNodeInfo.GetSize();
	for (int i=0;i<count;i++)
	{
		if(m_ArrayNodeInfo[i].NodeNum == num)
		{
			m_ArrayNodeInfo.RemoveAt(i);
			return TRUE;
		}
	}
	return FALSE;
}