C. Not Assigning (dfs)

传送门

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

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

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值