CF-1339D-D. Edge Weight Assignment

题目

题意：

n个节点，n-1条边，构成的树，让任意一对叶子节点，他两之间的所有边权异或值为0，边权可以用任意数字赋值，问最少使用多少个不同的数字，最多使用多少个不同的数字，可以满足上述条件。

思路：

先说最少的：

• 叶子节点之间距离，如果全是偶数，那边权就可以都为1

• 叶子节点之间距离，如果有奇数，那边权1 、2、3 这三个数就足够了
  最多的：

• 其实不管奇偶，我们凑二进制就行，比如 2⁰ 2¹ 2² 2³ ，凑起来这些数字，然后最后一个边（也就是到叶子节点的边）让一个二进制上每位都是 1 的数字来终结这些数字 就行，但是我们要考虑一个问题就是，比如这个叶子节点的兄弟节点也是叶子节点，这样这样答案就会减1；

代码：

#include<bits/stdc++.h>
#define ll long long
#include<vector>
using namespace std;
const int maxn=1e6+177;
vector<int >ve;
struct Node{
    int next;
    int to;
    int vul;
}edge[maxn];
int head[maxn];
int du[maxn];
int vis[maxn],dis[maxn];
int vv[maxn];
int tot;
int n;
void add(int from,int to,int vul){
    edge[++tot].next=head[from];
    edge[tot].to=to;
    edge[tot].vul=vul;
    head[from]=tot;
}

void init(){
    tot=0;
    memset(head,0,sizeof(head));
    memset(vis,0,sizeof(vis));
    memset(du,0,sizeof(du));
    memset(dis,1,sizeof(dis));
    memset(vv,0,sizeof(vv));
}


void dfs(int x,int fa){
    for(int i=head[x];i;i=edge[i].next){
        int y=edge[i].to;
        if(y==fa){
            continue;
        }
        dis[y]=dis[y]+dis[x];
        dfs(y,x);
    }
}


int main(){


    cin>>n;
    init();

    for(int i=1;i<=n;i++){
        dis[i]=1;
    }
    int x,y;
    for(int i=0;i<n-1;i++){
        cin>>x>>y;
        add(x,y,1);
        add(y,x,1);
        du[x]++;
        du[y]++;
    }
    int st;
    for(int i=1;i<=n;i++){
        if(du[i]==1){
           st=i;
           vv[i]=1;
        }

    }
    dis[st]=0;
    dfs(st,-1);
    int ans1=1;
    for(int i=1;i<=n;i++){
        if(vv[i]==1&&dis[i]%2==1){
            ans1=3;
        }
    }
    int ans2=n-1;
    for(int i=1;i<=n;i++){
        if(vv[i]==1){
            vv[i]=0;
            int jkl=edge[head[i]].to;
         //   cout<<jkl<<"   ##"<<endl;
            int tp=0;
            for(int j=head[jkl];j;j=edge[j].next){
                int y=edge[j].to;
         //       cout<<y<<"%%%%%%%%   "<<endl;
                if(vv[y]==1){
                    tp++;
                    vv[y]=0;
                }
            }
            ans2=ans2-tp;

        }

    }
    cout<<ans1<<" "<<ans2<<endl;
}