BZOJ 1086: [SCOI2005]王室联邦

2020-02-22 14:03:39 阅读：317 来源： 互联网

标签：ccnt 1086 head int cnt addd SCOI2005 define BZOJ

维护一个栈，从任意一个节点开始 dfs，在回溯时加入到栈中。
dfs每一个子树之后，若相对于刚进入该节点时栈的大小，新增加的节点超过 B，就将它们分为一块
这样分完保证不超过 2B，而整个dfs结束之后，若栈中还有节点，就把它们归为最后一块

#include <bits/stdc++.h>
#define pb push_back
#define fi first
#define se second
#define pii pair<int, int>
#define lp p << 1
#define rp p << 1 | 1
#define mid ((l + r) >> 1)
#define ll long long
#define db double
#define rep(i,a,b) for(int i=a;i<b;i++)
#define per(i,a,b) for(int i=b-1;i>=a;i--)
#define Edg int ccnt=1,head[N],to[N*2],ne[N*2];void addd(int u,int v){to[++ccnt]=v;ne[ccnt]=head[u];head[u]=ccnt;}void add(int u,int v){addd(u,v);addd(v,u);}
#define Edgc int ccnt=1,head[N],to[N*2],ne[N*2],c[N*2];void addd(int u,int v,int w){to[++ccnt]=v;ne[ccnt]=head[u];c[ccnt]=w;head[u]=ccnt;}void add(int u,int v,int w){addd(u,v,w);addd(v,u,w);}
#define es(u,i,v) for(int i=head[u],v=to[i];i;i=ne[i],v=to[i])
const int MOD = 1e9 + 7;
void M(int &x){if(x>=MOD)x-=MOD;if(x<0)x+=MOD;}
int qp(int a,int b=MOD-2){int ans=1;for(;b;a=1LL*a*a%MOD,b>>=1)if(b&1)ans=1LL*ans*a%MOD;return ans%MOD;}

char buf[1 << 21], *p1 = buf, *p2 = buf;
inline char getc() {
    return p1 == p2 && (p2 = (p1 = buf) + fread(buf, 1, 1 << 21, stdin), p1 == p2) ? EOF : *p1++;
}
inline int _() {
    int x = 0, f = 1; char ch = getc();
    while (ch < '0' || ch > '9') { if (ch == '-') f = -1; ch = getc(); }
    while (ch >= '0' && ch <= '9') { x = x * 10ll + ch - 48; ch = getc(); }
    return x * f;
}
const int N = 1111;
Edg
int n, b, st[N], top, belong[N], province[N], cnt;

void dfs(int u, int f) {
    int cur = top;
    es (u, i, v) {
        if (v == f) continue;
        dfs(v, u);
        if (top - cur >= b) {
            province[++cnt] = u;
            while (top != cur) belong[st[top--]] = cnt;
        }
    }
    st[++top] = u;
}

int main() {
    n = _(), b = _();
    rep (i, 1, n) {
        int u = _(), v = _();
        add(u, v);
    }
    dfs(1, 0);
    while (top) belong[st[top--]] = cnt;
    printf("%d\n", cnt);
    rep (i, 1, n + 1) printf("%d%c", belong[i], " \n"[i == n]);
    rep (i, 1, cnt + 1) printf("%d%c", province[i], " \n"[i == cnt]);
    return 0;
}

标签：ccnt,1086,head,int,cnt,addd,SCOI2005,define,BZOJ
来源： https://www.cnblogs.com/Mrzdtz220/p/12345169.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

ICode9

BZOJ 1086: [SCOI2005]王室联邦