POJ 2513Colored Sticks 并查集，字典树

本题综合性很强，用并查集是必须要用路劲压缩查找，其实可以用DFS来判断图的联通性，但肯定没并查集的效率高。字典树我用的是全局变量申请，但要开足够大的空间，开500000RUN ERROR一次，结果开500000*10就A的，而且时间只有 600MS ，字典树也可以动态创建，但时间肯定要多花的，时间空间自己去取舍吧..

#include <iostream>

using namespace std;

#define  ST_COUNT  500000
struct TNode
{
int parent;
int data;
}sn[ST_COUNT];

struct TrieNode
{
int data;
int child[26];
}trNode[ST_COUNT*10]; // 字典树节点要开多一点
int tr_index = 1;

int g_NodeId = 0; // 给每一个节点的唯一编号
int NodeDegree[ST_COUNT]; // 存放节点的度

int CreateTrieNode(int inData)
{
memset(trNode[tr_index].child, 255, sizeof(trNode[tr_index].child) );
trNode[tr_index].data = inData;
tr_index++;
return tr_index-1;
}

int FindAddInsertTrie(char szStr[], int szLen) // 查询字典树，并插入新的节点
{
int trBase = 0, re, tdata;
int i;
for(i = 0; i < szLen; i++)
{
if( trNode[trBase].child[szStr[i] - 'a'] == -1 )
{
if( szLen-i-1 == 0 ) tdata = g_NodeId++;
else tdata = -1;
re = CreateTrieNode(tdata);
trNode[trBase].child[szStr[i] - 'a'] = re;
trBase = re;
}
else
{
trBase = trNode[trBase].child[szStr[i] - 'a'];
}
}

return trNode[trBase].data;
}

int findPEx(int inf) // 路径压缩查找
{
return sn[inf].parent == -1 ? inf : sn[inf].parent = findPEx(sn[inf].parent);
}

int findP(int inf)
{
int tf;
if( sn[inf].parent == -1 )
{
return inf;
}
else
{
tf = findP(sn[inf].parent);
sn[inf].parent = tf;
return tf;
}
}
int main()
{
// 61088K  594MS
freopen("in.txt", "r", stdin);

int i, j, ta, tb, fa, fb;
char sza[12], szb[12];
// 计算节点的度
// 判断图的联通性
memset(trNode[0].child, 255, sizeof(trNode[0].child) );
trNode[0].data = -1;
memset(NodeDegree, 0, sizeof(NodeDegree) );
for(i = 0; i < ST_COUNT; i++) sn[i].parent = -1;

while( scanf("%s", sza) != EOF )
{
scanf("%s/n", szb);
ta = FindAddInsertTrie(sza, strlen(sza) ); // 查找字典树，并插入
tb = FindAddInsertTrie(szb, strlen(szb) );

NodeDegree[ta]++; // 计算节点的度
NodeDegree[tb]++;

fa = findPEx(ta); // 查找集合
fb = findPEx(tb);
if( fa != fb ) // 合并
{
sn[fa].parent = fb;
}
}

int odd = 0;
bool IsLink = true, IsO = true;
int only = findPEx(0);
for(i = 1; i < g_NodeId; i++) // 判断图的联通性
{
if( only != findPEx(i) )
{
IsLink = false;
break;
}
}

if( IsLink )
{
for(i = 0; i < g_NodeId; i++) // 判断欧拉路
{
if( NodeDegree[i]%2 != 0 ) odd++;
if( odd > 2 ) break;
}
if( odd > 2 )
cout<<"Impossible"<<endl;
else
cout<<"Possible"<<endl;
}
else
cout<<"Impossible"<<endl;

return 0;
}