【Hash】dr所在国度

 

hash

Description

 

dr所在国度的有个奇怪的规定：他们的字母不是a~z，而是用1~1000表示。

利用这个奇怪的规定，dr想出了一个好玩的游戏：首先给出n个字符串（当然每个字符用1~1000表示），然后给出有m个节点的树，节点编号1~m，这棵树以1号节点为根，每个节点都包含一个字符。现在要求用从根节点到其他m-1个节点的链上的字符组成m-1个新字符串（字符的排列顺序为从根到终点的顺序）。

是否这m-1个新字符串中的任意一个串，都与给出的n个字符串中至少一个串匹配呢？

字符串S与字符串T匹配：设len1=|S|,len2=|T|len1=∣S∣,len2=∣T∣，则S_{1}=T_{1},S_{2}=T_{2},...,S_{len1}=T_{len1}S1​=T1​,S2​=T2​,...,Slen1​=Tlen1​ 且 len1 \leq len2len1≤len2

Input

 

第一行输入nn和mm，表示有nn个字符串，树的节点数为m(n \leq 10^3,m \leq 2 * 10^5)m(n≤103,m≤2∗105)

接下来nn行，每行首先输入一个整数kk，代表字符串长度，然后输入kk个整数，代表字符串S_{i}(|Si|,\sum |Si| \leq 2 * 10^5)Si​(∣Si∣,∑∣Si∣≤2∗105)

接下来一行，输入mm个整数C_{i}Ci​，表示树上第ii个节点上的字符

接下来m-1m−1行，每行输入2个整数u,v(1 \leq u,v \leq m)u,v(1≤u,v≤m)，表示uu和vv有一条边

Output

 

输出占一行，都能匹配输出YES，否则输出NO

Sample Input 1 

1 4
4 1 2 3 4
1 2 3 2
1 2
2 3
1 4

Sample Output 1

YES

 

#include<cstdio>
#include<algorithm>
#include<map>
#include<cstring>
using namespace std;
typedef long long ll;

map<ll,int>mp;
ll Hash=9731;
int w[200005];

struct node
{
    int v,nxt;
}edge[400005];
int head[400005],ei;

void addedge(int u,int v)
{
    edge[ei].v=v;
    edge[ei].nxt=head[u];
    head[u]=ei++;
}

void init()
{
    memset(head,-1,sizeof(head));
    ei=0;
}
int flag=0;
bool vis[200005]={0};
void dfs(int x,int deep,ll ans,ll sed)
{
    ans+=w[x]*sed;
    sed*=Hash;
    vis[x]=1;
    //printf("%lld ",ans);
    if(mp[ans]==0)flag=1;
    for(int i=head[x];i!=-1;i=edge[i].nxt)
    {
        int v=edge[i].v;
        if(vis[v]==1)continue;
        dfs(v,deep+1,ans,sed);
    }
}

int main()
{
    int n,m;
    init();
    scanf("%d%d",&n,&m);
    for(int i=0;i<n;i++)
    {
        int t;
        scanf("%d",&t);
        ll h=0;
        ll sed=Hash;
        for(int j=0;j<t;j++)
        {
            int x;
            scanf("%d",&x);
            h+=x*sed;
            sed*=Hash;
            mp[h]=1;
        }
    }
    for(int i=1;i<=m;i++)
    {
        scanf("%d",&w[i]);
    }
    for(int i=0;i<m-1;i++)
    {
        int u,v;
        scanf("%d%d",&u,&v);
        addedge(v,u);
        addedge(u,v);
    }
    dfs(1,0,0,Hash);
    if(flag==0)printf("YES\n");
    else printf("NO\n");
}