Problem - 1627C - Codeforces
题意:
一个n个节点的数, 给出 n-1 条边, 要求给每一条边赋值,使其满足:任意一条边或相连的两条边之和都是质数
题解:
这种情况,只有2和其它质数相加也是质数,这种题还是比较多, 能较容易想到。其次,题目没说所用的数不能重复, 索性只用2 3好了。 2333
接下来就是遍历树了,题目也没有说遍历头一定是1哦,最近好喜欢用map ^-^ , 好久没做链表加dfs来看路了, 竟然卡了
上代码:
#include <bits/stdc++.h> using namespace std; typedef long long LL; typedef pair<int,int> PII; const int N = 2e5+10; int ne[N], h[N], e[N], idx=0; int n, num[N],res[N]; bool st[N]; map<PII, int> mp; void add(int a, int b) { e[idx] = b, ne[idx] = h[a],h[a]=idx++; } void dfs(int x, int nn) { st[x] = 1; for(int i = h[x]; i!=-1; i = ne[i]) { if(!st[e[i]]) mp[{x,e[i]}]=nn, dfs(e[i], 5-nn); } } int main() { int t; cin >> t; while(t --) { memset(h, -1, sizeof h); memset(num, 0, sizeof num); memset(st, 0, sizeof st); mp.clear(); vector<PII> ve; cin >> n; bool f=0; for(int i = 1; i < n; i ++) { int a, b; cin >> a >> b; if(a>b)swap(a,b); add(a, b); add(b, a); ve.push_back({a,b}); ve.push_back({b,a}); num[a]++; num[b]++; if(num[a]>2 || num[b]>2)f=1; } if(f) { puts("-1"); continue; } for(int i=1;i<=n;i++) if(num[i]==1) { dfs(i,2); break; } for(int i =0; i < ve.size(); i ++) if(mp[ve[i]]) cout << mp[ve[i]]<<' '; cout<<'\n'; } return 0; }
标签:idx,--,质数,++,st,链表,int,num 来源: https://www.cnblogs.com/la-la-wanf/p/16187856.html
本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享; 2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关; 3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关; 4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除; 5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。