bzoj1086: [SCOI2005]王室联邦

这道题有点坑。。
不用输出最优解。。
只需输出一组解。。
用一个栈来存当前省的城市。。
一个暴搜O(n)的算法就可以过。。
代码挺简单的。。

#include<cstdio>#include<cstring>#include<iostream>using namespace std;struct node {    int x,y,next;}a[2100];int len,last[1100];void ins(int x,int y) {    len++;    a[len].x=x;a[len].y=y;    a[len].next=last[x];last[x]=len;}int top,m,sta[1100],n;int B,d[1100],g[1100];void dfs(int x,int fa) {    int tt=top;    for(int k=last[x];k;k=a[k].next) {        int y=a[k].y;        if(y!=fa) {            dfs(y,x);            if(top-tt>=B) {                g[++m]=x;                while(top!=tt)                    d[sta[top--]]=m;            }        }    }    sta[++top]=x;}int main() {    scanf("%d%d",&n,&B);    len=0;memset(last,0,sizeof(last));    for(int i=1;i<n;i++) {        int x,y;scanf("%d%d",&x,&y);        ins(x,y);ins(y,x);    }    m=0;top=0;    memset(sta,0,sizeof(sta));    dfs(1,0);    while(top!=0)        d[sta[top--]]=m;    printf("%d\n",m);    for(int i=1;i<n;i++)        printf("%d ",d[i]);    printf("%d\n",d[n]);    for(int i=1;i<m;i++)        printf("%d ",g[i]);    printf("%d\n",g[m]);    return 0;}