题意:给你一棵树然后让你构造树的边权,使每个点的任意两边(如果没有两边就只算一边)的边权之和为质数。
思路:看了很久没看明白是啥意思,然后就发现如果是要构成满足条件的树的话,那么树的每个度绝对不能超过2(大于等于三并且满足条件的树推不出来),那么满足条件的树就相当于是一个链表,可以直接从度为1的地方开始,然后一直dfs,然后是质数可以直接用二进制的(10和11,也就是2和3来不断取答案)。最后就是初始化的时候如果memset用的多会直接t3?
代码
/**
* ┏┓ ┏┓+ +
* ┏┛┻━━━┛┻┓ + +
* ┃ ┃
* ┃ ━ ┃ ++ + + +
* ████━████+
* ◥██◤ ◥██◤ +
* ┃ ┻ ┃
* ┃ ┃ + +
* ┗━┓ ┏━┛
* ┃ ┃ + + + +Code is far away from
* ┃ ┃ + bug with the animal protecting
* ┃ ┗━━━┓ 神兽保佑,代码无bug
* ┃ ┣┓
* ┃ ┏┛
* ┗┓┓┏━┳┓┏┛ + + + +
* ┃┫┫ ┃┫┫
* ┗┻┛ ┗┻┛+ + + +
*/
#include<cstdio>
#include <iostream>
#include <algorithm>
#include <string.h>
#include <string>
#include <math.h>
#include<vector>
#include<queue>
#include<map>
#define sc_int(x) scanf("%d", &x)
#define sc_ll(x) scanf("%lld", &x)
#define pr_ll(x) printf("%lld", x)
#define pr_ll_n(x) printf("%lld\n", x)
#define pr_int_n(x) printf("%d\n", x)
#define ll long long
using namespace std;
const int N=1000000+100;
int n ,m;
ll s[N],d[N];
int ne[N],e[N],idx,h[N];
bool st[N];
map<pair<int,int>,int>q;
void dfs(int x,int co)
{
for(int i =h[x];i!=-1;i=ne[i])
{
int j=e[i];
if(!st[j]){
st[j]=1;
q[{x,j}]=q[{j,x}]=co;
dfs(j,co^1);
}
}
}
void add(int a,int b)
{
ne[idx]=h[a];
e[idx]=b;
h[a]=idx++;
}
void init(int x)
{
for(int i =1;i<=x;i++)
{
h[i]=-1;
st[i]=0;
ne[i]=0;
e[i]=0;
d[i]=0;
}
}
void solve()
{
q.clear();
cin>>n;
init(n);
vector<pair<int,int>>c;
for(int i =1;i<n;i++)
{
int a,b;
cin>>a>>b;
add(a,b),add(b,a);
c.push_back({a,b});
d[a]++,d[b]++;
}
int k=0;
for(int i =1;i<=n;i++)
{
if(d[i]>2){
k=-1;
break;
}
else if(d[i]==1)
k=i;
}
if(k==-1){
cout<<"-1\n";
return ;
}
// cout<<k<<endl;
st[k]=1;
dfs(k,2);
//cout<<" ?";
for(int i =0;i<c.size();i++)
{
int a=c[i].first,b=c[i].second;
cout<<q[{a,b}]<<" ";
}
cout<<endl;
return ;
}
int main()
{
int t;
sc_int(t);
while(t--)solve();
return 0;
}