tries树学习 poj 2513

参考了：http://wenku.baidu.com/view/7a5ef4eb6294dd88d0d26b88.html的解题报告

题目大意：是给一组单词，每组单词两个，表示棍子的两头的颜色，要是两根棍子的两头有颜色相同的，则表示这两根棍子可以连接在一起

要求你来判断这些棍子能否都连接在一起

解题思路：这题用到的知识有欧拉图，tries树和并查集

可以把棍子两头的颜色看成是两个节点，两个节点相连，相同颜色的就成为一个公共点

例如样例就可以建成下面这个图

那么判断棍子是否都能连接起来，只要判断这个图是否可以一笔划，覆盖所有的点

即判断这个图是否是个欧拉图，满足欧拉图的条件是：图为连通图，然后每个结点的度数要么都为偶数，要么只有两个节点的度数为奇数

用tries树来保存每个颜色的单词，通过tries来为每个颜色单词标上序号，然后根据这个序号用并查集来合并一个棍子上的两个颜色节点

在进行tries树插入单词时，返回单词的序号，并记录单词出现的次数，即为这个单词节点的度数，用来判断欧拉图中节点度数是否满足情况

并查集用来合并两个节点，查找函数用来在合并完后判断每个结点的是否在一个集合内，用来判断图是否连通

 

#include <iostream>
#include <cstdio>
#include <cstring>

using namespace std;

const int maxn = 500001;
const int len = 26;

typedef struct triesNode
{
	int n;  //颜色序号或标志，标志是否为单词
	triesNode *next[len];
}TriesTree;   //定义一棵Trie树



int num=0;   //颜色序号
int degree[maxn]; //a数组用来存放每种颜色出现的次数，即该点度数
int parent[maxn];  //存放连通分量号,即根结点序号
//int count_f[maxn];  //存放各颜色点所在的连通分量包含的结点数

int insert(char *str, TriesTree* head);
void unit(int x, int y);
int find(int x);
bool judge(); // 判断结点度数是否符合欧拉图的性质
bool isConnected(); //判断是否连通

int main()
{
    
    char str1[20],str2[20];
	int x, y, i;
	TriesTree *head = new TriesTree();
	head->n = -1;
	memset(head->next, 0, sizeof(head->next));
	memset(degree, 0 ,sizeof(degree));
	for(i = 0; i < maxn; i++)
		parent[i] = i;
	while(scanf("%s %s", str1, str2) != EOF)
	{
		x = insert(str1, head);
		y = insert(str2, head);
		degree[x]++;
		degree[y]++;
		unit(x, y);
	}
	if(!isConnected() || !judge())
        printf("Impossible\n");
    else
        printf("Possible\n");
    
    return 0;
}

int insert(char *str, TriesTree* head)
{
    int i, length = strlen(str);
    TriesTree *p = head;
    for(i = 0; i < length; i++)
    {
        if(!p->next[str[i] - 'a'])
            break;
        p = p->next[str[i] - 'a'];
    }
    
    while(i < length)
    {
        TriesTree* node = new TriesTree();
        node->n = -1;
        memset(node->next, 0, sizeof(node->next));
        p->next[str[i] - 'a'] = node;
        p = node;
        i++;
    }
    if(p->n == -1)
        p->n = num++;
    return p->n;
}

int find(int x)
{
    if(x != parent[x])
        parent[x] = find(parent[x]);
    return parent[x];
}

void unit(int x, int y)
{
    int px, py;
    px = find(x);
    py = find(y);
    if(px != py)
        parent[py] = px;
}

bool judge()
{
    int val = 0;
    for(int i = 0; i < num; i++)
    {
        if(degree[i] & 1 != 0)
            val++;
    }
    if(val != 0 && val != 2)
        return false;
    return true;
}

bool isConnected()
{
    int root = find(0);
    for(int i = 1; i < num; i++)
    {
        if(find(i) != root)
            return false;
    }
    return true;
}