Colored Sticks--木棒的颜色--并查集+字典树+哈希

JdiLfc

于 2020-11-14 14:48:02 发布

阅读量422

点赞数

分类专栏： ACM刷题并查集

本文链接：https://blog.csdn.net/JdiLfc/article/details/109690762

版权

ACM刷题同时被 2 个专栏收录

56 篇文章 0 订阅

订阅专栏

并查集

4 篇文章 0 订阅

订阅专栏

该程序实现了一种算法，通过并查集与字典树结合，判断给定的木棒首尾颜色是否能连成一根。输入是不同木棒的首尾颜色对，输出是可能或不可能。程序首先初始化所有节点，然后根据输入的颜色对更新字典树和并查集，最后检查是否存在欧拉路径来确定答案。

摘要由CSDN通过智能技术生成

给出木棒的首尾颜色，看能不能连成一根。

考察并查集+字典树+哈希

Sample Input

blue red
red violet
cyan blue
blue magenta
magenta cyan

Sample Output

Possible

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
int degree[500010],fa[500010],id = 1;

struct node
{
    int flag;
    int id;
    node* nxt[26];
    node ()
    {
        for(int i=0;i<26;i++)
            nxt[i]=NULL;
        flag=0;
    }
};
node* root;

int fid(int x)
{
    if(fa[x] != x)
        fa[x] = fid(fa[x]);//路径压缩；
    return fa[x];
}

//字典树哈希
int Hash(char s[])
{
    node *p = root;
    for(int i = 0; s[i]; i++)
    {
        if(p->nxt[s[i]-'a'] == NULL)
            p->nxt[s[i]-'a'] = new node();
        p = p->nxt[s[i]-'a'];
    }
    if(p->flag != 1)
    {
        p->flag = 1;
        p->id = id++;
    }
    return p->id;
}

int check()
{
    int sum = 0;
    int x = fid(1);
    for(int i = 2; i < id; i++)
        if(fid(i) != x)//没有共同祖先，图是不连通的，直接返回；
            return 0;
    for(int i = 1; i < id; i++)
    {
        if(degree[i]%2)
            sum++;
    }
    if(sum == 0 || sum == 2)
        return 1;//图是连通的并且奇度数是0或2，说明有欧拉路；
    return 0;//图是连通的但奇度数不是0或2也不存在欧拉路；
}

int main()
{
    memset(degree,0,sizeof(degree));
    for(int i = 1; i <= 500000; i++)
        fa[i] = i;//所有节点初始化为一棵树
    char s1[12],s2[12];
    int u,v;
    root = new node();
    while(scanf("%s %s",s1,s2) != EOF)
    {
        u = Hash(s1);
        v = Hash(s2);
        degree[u]++;
        degree[v]++;
        int x = fid(u);
        int y = fid(v);
        if(x != y)
            fa[x] = y;
    }
    if(check()) printf("Possible\n");
    else printf("Impossible\n");
    return 0;
}