#3702. 月读(tsukuyomi)

题目描述
有些时候,出题人真的不想写背景。

总而言之,月读现在有一棵大小为 $N$ ,树上每条边上有一个数字,月读有 $M$ 次询问,每次询问一对 $(x,y)$ ,你需要回答从 $x$ 到 $y$ 的路径上的数字重新排列能否形成一个回文序列,若可行输出 $Yes$ ,否则输出 $No$ ,月读为了加快您的读入,每次询问的 $x,y$ 是通过某种方式生成的,为了加快您的输出,你只需要最后输出回答 $Yes$ 的个数和即可。
数据范围
$N \le 10^6,M \le 10^7$ ,对于所有的 $w_i \le n$
题解
考虑到如果能够重新排列,使其为回文序列的话,则满足最多一个数字出现的次数是奇数

对每种边权进行 $hash$ 赋值,和 $0$ 一起扔进 $hash$ 表里,然后记 $f_i$ 为根到 $i$ 点的边权异或和,对于询问的点 $x,y$ ,若 $f_x\ \oplus\ f_y$ 的值在 $hash$ 表里出现过,则为合法的

$hash$ 很随机,所以可过,效率 $O(n+m)$
代码

#include <bits/stdc++.h>
#define _(d) while(d(isdigit(c=getchar())))
typedef unsigned long long U;
using namespace std;const U B=793999;
const int N=1e6+5,K=1e6+7,M=N<<1;
int Rd(){
    char c;_(!);int x=c^48;_()x=(x<<3)+(x<<1)+(c^48);return x;
}
struct Hash{
    int hd[K],t,V[N],nx[N];U W[N];
    void add(U x){
        int y=x%K;
        nx[++t]=hd[y];W[hd[y]=t]=x;
    }
    bool find(U x){
        int y=x%K;
        for (int i=hd[y];i;i=nx[i])
            if (W[i]==x) return 1;
        return 0;
    }
}hs;
int a,b,m,n,t,hd[N],V[M],W[M],nx[M],c,e[N],ans;
U g[N],h[M];
void add(int u,int v,int w){
    nx[++t]=hd[u];V[hd[u]=t]=v;W[t]=w;
}
void dfs(int u,int fr,U lst){
    h[e[u]=++c]=lst;
    for (int i=hd[u];i;i=nx[i])
        if (V[i]!=fr)
            dfs(V[i],u,g[W[i]]),
            h[++c]=g[W[i]];
}
int main(){
    n=Rd(),m=Rd();g[0]=1;hs.add(0);
    for (int i=1;i<=n;i++)
        g[i]=g[i-1]*B,hs.add(g[i]);
    for (int u,v,w,i=1;i<n;i++)
        u=Rd(),v=Rd(),w=Rd(),
        add(u,v,w),add(v,u,w);dfs(1,0,(U)0);
    for (int i=2;i<=c;i++) h[i]^=h[i-1];
    a=Rd();b=Rd();
    for (int x,y;m--;){
        x=a%n+1;y=b%n+1;
        a=666073ll*a%1000000007;
        b=233ll*b%998244353;
        ans+=hs.find(h[e[x]]^h[e[y]]);
    }
    return printf("%d\n",ans),0;
}

 

posted @ 2019-08-05 20:27  xjqxjq  阅读(265)  评论(0编辑  收藏  举报