先染色,一条边连接的俩点染成不同颜色(好像叫二部图?)
然后,每一对与距离3有不同的颜色!
设R为红色顶点的数量,B为蓝色顶点的数量。我们将假设
R≤B。
设X是不超过N/3的最大整数。有两种情况。
当R≤X时
将3的倍数分配给红色顶点。因为每一对都有一个红色的顶点,它们的乘积
值总是3的倍数。
当R > X时
将所有3k + 1类型值分配给红色顶点,并将3k + 2类型值分配给蓝色顶点。
这样,对于每一对,会满足其中一个条件:
1 其中一个顶点是3的倍数。那么这一对乘积是3的倍数。
2 一个顶点的值是1 mod 3,另一个是2 mod 3。在这种情况下,和是3的倍数
#include<bits/stdc++.h>
using namespace std;
const int maxn=2e5+7;
vector<int> g[maxn];
int co[maxn];
int ans[maxn];
int use[maxn];
vector<int> r;
vector<int> b;
void dfs(int now,int x){
co[now]=x;
for(int i=0;i<g[now].size();i++){
int u=g[now][i];
if(co[u]==0) dfs(u,x*-1);
}
}
int main(){
int n;
cin>>n;
int t=n-1;
while(t--){
int a,b;
cin>>a>>b;
g[a].push_back(b);
g[b].push_back(a);
}
dfs(1,1);
for(int i=1;i<=n;i++){
if(co[i]==1) r.push_back(i);
if(co[i]==-1) b.push_back(i);
}
if(r.size()>b.size()) swap(r,b);
if(r.size()>n/3){
int i=0;
for(int k=0;3*k+1<=n;k++){
ans[r[i++]]=3*k+1;
use[3*k+1]=1;
}
i=0;
for(int k=0;3*k+2<=n;k++){
ans[b[i++]]=3*k+2;
use[3*k+2]=1;
}
int k=1;
for(int i=1;i<=n;i++){
if(ans[i]==0){
ans[i]=3*k;k++;
}
}
}
else{
int k=1;
for(int i=0;i<r.size();i++){
ans[r[i]]=3*k;
use[3*k]=1;
k++;
}
k=1;
for(int i=1;i<=n;i++){
if(ans[i]==0){
while(use[k]==1){
k++;
}
ans[i]=k;
use[k]=1;
}
}
}
for(int i=1;i<=n;i++) cout<<ans[i]<<" ";
return 0;
}