Codeforces Round # 746-Div. 2-C Bakry and Partitioning题解

题目链接

题目链接放在Bakry and Partitioning，大意就是给一棵有$n$个节点的树，树上每个节点$i$都带有一个权值$a_i$，现在要把这棵树分成满足条件的x个联通分支（2$⩽x⩽k,k$为题目给定的值）：每个联通分支所有节点的权值做异或运算后得到一个值而且所有$x$个值要相等。如果存在满足条件的分法则输出$YES$，否则输出$NO$。

题目分析

先说下题目思路，由异或的运算规则可以得到两条性质：

1. $\forall x\in \mathbf{N},x^{(n)}:=x\mathrm{x}\mathrm{o}\mathrm{r}x\cdots \mathrm{x}\mathrm{o}\mathrm{r}x$(一共$n$个$x$)，则当$n$为偶数时，$x^{(n)}=0$；当$n$为奇数时，$x^{(n)}=x$.
2. $x\mathrm{x}\mathrm{o}\mathrm{r}0=x$.

由此我们可以得到，当给定的数可被分为偶数个满足条件的联通分支时，所有节点的异或值为0（充要条件）。我们只需要将树分为任意的两个联通分支即可；当给定给定的数只可被分为奇数个满足条件的联通分支时，此时设所有节点的异或值为$x(x\ne 0)$，我们在树中需要找到两支互不相交的异或值均为$x$的联通分支，然后将它们与树的其余部分分开，从而形成三个异或值为$x$的联通分支，如果找不到的话说明没有满足题目要求的分法。
具体实现步骤如下：

1. 读取数据并建图，然后对所有节点的权值求异或，最后得到值$x$，转步骤2.
2. 如果$x=0$，则直接输出$YES$，转步骤1；否则转步骤3.
3. 如果$k=2$，则直接输出$NO$，转步骤1；否则转步骤4.
4. 以$1$为根节点对树进行$DFS$，在遍历同时寻找互不相交的异或值等于$x$的联通分支个数$num$.如果$num>=2$，则输出$YES$,转步骤1；否则输出$NO$，转步骤1.

源代码

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
using namespace std;
const int N=1e5+5;
int t,n,k,a[N],tree[N],head[N],to[N*2],cnt,num,xorall=0;
bool vis[N];
vector<int>nxt(N<<1,0);
void X_OR(int x,int pre)
{
    tree[x]=a[x];
    for(int i=head[x];i;i=nxt[i])
    {
        int &nod=to[i];
        if(nod==pre) continue;
        X_OR(nod,x);
        tree[x]^=tree[nod];
    }
    if(tree[x]==xorall) num++,tree[x]=0;
    return;
}
inline void addedge(int u,int v)
{
    nxt[++cnt]=head[u];
    head[u]=cnt;
    to[cnt]=v;
    return;
}
int main()
{
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d",&n,&k);
        memset(head,0,sizeof head);
        nxt.push_back(0);
        cnt=num=xorall=0;
        for(int i=1;i<=n;i++) scanf("%d",a+i),xorall^=a[i];
        for(int i=1;i<n;i++)
        {
            int u,v;
            scanf("%d%d",&u,&v);
            addedge(u,v);
            addedge(v,u);
        }
        if(!xorall) {printf("YES\n");continue;}
        X_OR(1,0);
        nxt.clear();
        if(k==2) printf("NO\n");
        else if(num>1) printf("YES\n");
        else printf("NO\n");
    }
    return 0;
}

吐槽

在比赛中拿到这个题时我本来打算用并查集做的后来发现并不行，然后就白白的浪费了时间；在比赛结束后，我试着开普通数组做这道题，结果ac了但耗费了600多ms,后来才发现memset函数非常耗时！！！ 一般对于$10^5\sim 10^6$这种长度的数组使用$memset$函数时，耗时已经非常可观了，为此我思考了一下，决定取消每次对链式前向星的$to$数组使用该函数，然后把$nxt$数组改成了$vector$，这才把时间降到不到300ms。至于$head$数组我没想好怎么改，如果有大神能指点的话感激不尽。