poj 2513 Colored Sticks(Trie+hash+dsu)

题目链接：http://poj.org/problem?id=2513

题意：n个木棍，木棍两个端点分别涂上色，问能否将所有木棍都连接起来，要求是木棍连接的两个端点颜色必须相同。

思路：这题用STL会超时，否则可以直接用map+dsu，所以只能用hash，用字典树作出string'到int的映射，然后用并查集判断欧拉回路。

无向图判断欧拉回路的条件是：

①所有顶点的度数均为偶数（这道题连接之后的木棍两个端点度数为1）

②图是连通的（可用并查集判断，所有顶点只有一个公共祖先即符合条件）

（话说并查集真是个神奇的东西..）

poj 2315：

#include <limits.h>
#include <math.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <time.h>
#include <algorithm>
#include <iostream>
#include <iterator>
#include <sstream>
#include <queue>
#include <stack>
#include <string>
#include <vector>
#include <list>
#include <set>
//#define ONLINE_JUDGE
#define eps 1e-6
#define INF 0x7fffffff                                          //INT_MAX
#define inf 0x3f3f3f3f                                          //int??????????????????
#define FOR(i,a) for((i)=0;i<(a);(i)++)                          //[i,a);
#define MEM(a) (memset((a),0,sizeof(a)))
#define sfs(a) scanf("%s",a)
#define sf(a) scanf("%d",&a)
#define sfI(a) scanf("%I64d",&a)
#define pf(a) printf("%d\n",a)
#define pfI(a) printf("%I64d\n",a)
#define pfs(a) printf("%s\n",a)
#define sfd(a,b) scanf("%d%d",&a,&b)
#define sft(a,b,c)scanf("%d%d%d",&a,&b,&c)
#define for1(i,a,b) for(int i=(a);i<b;i++)
#define for2(i,a,b) for(int i=(a);i<=b;i++)
#define for3(i,a,b)for(int i=(b);i>=a;i--)
#define MEM1(a) memset(a,0,sizeof(a))
#define MEM2(a) memset(a,-1,sizeof(a))
#define MEM3(a) memset(a,0x3f,sizeof(a))
#define MEMS(a) memset(a,'\0',sizeof(a))
#define LL __int64
const double PI = acos(-1.0);
template<class T> T gcd(T a, T b) { return b ? gcd(b, a % b) : a; }
template<class T> T lcm(T a, T b) { return a / gcd(a, b) * b; }
template<class T> inline T Min(T a, T b) { return a < b ? a : b; }
template<class T> inline T Max(T a, T b) { return a > b ? a : b; }
using namespace std;
template<class T>
T Mint(T a, T b, T c) {
    if (a>b) {
        if (c>b)
            return b;
        return c;
    }
    if (c>a)
        return a;
    return c;
}
template<class T>
T Maxt(T a, T b, T c) {
    if (a>b) {
        if (c>a)
            return c;
        return a;
    }
    else if (c > b)
        return c;
    return b;
}

const int maxn=500010;
int T,n,m;
int f[maxn],degree[maxn];
int color;

int find(int x){//路径压缩
	return f[x]==-1?x:find(f[x]);
}

void Union(int a,int b){//合并
	int ra=find(a);
	int rb=find(b);
	if(ra==rb) return ;
	f[ra]=rb;
}

struct Trie{//构造字典树
	Trie *next[26];//单词指针
	int id;//颜色编号
	bool isword;//判断是否结束
};

Trie *root;

int hash(char *str){//利用字典树构造字符串到编号int的映射
	int len=strlen(str);
	Trie *p=root,*q;
	for1(i,0,len){
		int id=str[i]-'a';
		if(p->next[id]==NULL){
			q=(Trie*)malloc(sizeof(Trie));
			for1(j,0,26)
				q->next[j]=NULL;
			q->isword=false;
			q->id=0;
			p->next[id]=q;
		}
		p=p->next[id];
	}
	if(p->isword){//若颜色单词已经存在，返回其编号
		return p->id;
	}
	else{
		p->isword=true;
		p->id=color++;//若颜色单词不存在，分配编号并返回
		return p->id;
	}
}

void del(Trie *T){
	if(T==NULL) return ;
	for1(i,0,26){
		if(T->next[i]!=NULL)
			del(T->next[i]);
	}
	free(T);
}

int main(){
#ifndef ONLINE_JUDGE
    freopen("test.in","r",stdin);
    freopen("test.out","w",stdout);
#endif
    MEM2(f);
    MEM1(degree);
    root = (Trie *) malloc (sizeof(Trie));
    for1(i,0,26)//初始化根节点
    	root->next[i]=NULL;
    root->isword=false;
    root->id=0;
    color=0;//初始化颜色编号
	char s1[25],s2[25];
    int cnt1=0,cnt2=0;
    while(~scanf("%s%s",s1,s2)){
    	int t1=hash(s1);//记录s1,s2的颜色编号
    	int t2=hash(s2);
    	degree[t1]++;//记录s1,s2颜色编号出现的次数,即度数
    	degree[t2]++;
    	Union(t1,t2);//合并
    }
    for1(i,0,color){
    	if(f[i]==-1) cnt1++;
    	if(degree[i]&1) cnt2++;
    	if(cnt1>1) break;//若存在多个祖先，则图为森林，不连通
    	if(cnt2>2) break;//若度数为奇数的节点数>3，欧拉回路必不存在
    }
    if((cnt1==0||cnt1==1)&&(cnt2==0||cnt2==2)){//数据略坑，有空数据的情况
    	printf("Possible\n");
    }
    else{
    	printf("Impossible\n");
    }
    return 0;
}

.