C. Not Assigning （dfs）

最新推荐文章于 2025-12-19 15:31:04 发布

原创最新推荐文章于 2025-12-19 15:31:04 发布 · 188 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#深度优先 #算法

c++ 专栏收录该内容

85 篇文章

订阅专栏

本文介绍了一种构造树形结构的方法，使得任意相邻两条边的权重之和为质数。通过限制节点度数不超过2，将问题简化为链表形式，并采用深度优先搜索算法进行遍历赋值。

传送门

题意：给你一棵树然后让你构造树的边权，使每个点的任意两边（如果没有两边就只算一边）的边权之和为质数。

思路：看了很久没看明白是啥意思，然后就发现如果是要构成满足条件的树的话，那么树的每个度绝对不能超过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;
}