洛谷P2325王室联邦 SCOI2005 构造+树上分块

正解:构造

解题报告:

照例先放传送门

umm其实我jio得这题应该在教树上莫队的时候港,应该是用来帮助理解树上莫队的分块方式的

然而这题是在学了树上分块之后再遇到的?就显得没那么难了吼

然后就随便说下趴

就利用dfs出栈序,没了

,,,dbq我jio得我说得太简略了QAQ

但确实没什么好说的?可能作为一个构造题难点在于想到树上分块相关趴QwQ

over

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define rp(i,x,y) for(register ll i=x;i<=y;++i)
#define rpb(i,x) for(register ll i=head[x];i;i=edge[i].nxt)

const int N=+;
int head[N],tot,n,b,bt,bl[N],stk[N],sh[N],cnt;
struct ed{ll to,nxt;}edge[N<<];

inline ll read()
{
    register char ch=getchar();register ll x=;register bool y=;
    while(ch!='-' && (ch>'' || ch<''))ch=getchar();
    if(ch=='-')ch=getchar(),y=;
    while(ch>='' && ch<='')x=(x<<)+(x<<)+(ch^''),ch=getchar();
    return y?x:-x;
}
inline void ad(int x,int y){edge[++tot]=ed{y,head[x]};head[x]=tot;}
void dfs(int x,int fa)
{
    ll tp=bt;
    rpb(i,x)
    {
        ll tmp=edge[i].to;if(tmp==fa) continue;
        dfs(tmp,x);
        if(bt-tp>=b)
        {
            sh[++cnt]=x;
            while(bt>tp)bl[stk[bt--]]=cnt;
        }
    }
    stk[++bt]=x;
}
int main()
{
    n=read();b=read();
    rp(i,,n-){ll u=read(),v=read();ad(u,v);ad(v,u);}
    dfs(,);while(bt)bl[stk[bt--]]=cnt;
    printf("%d\n",cnt);
    rp(i,,n)printf("%d ",bl[i]);printf("\n");
    rp(i,,cnt)printf("%d ",sh[i]);
    return ;
}