luogu P2018 消息传递

二次联通门 : luogu P2018 消息传递

/*
    luogu P2018 消息传递

    树形dp
    原来用优先队列做了一下， T了俩点
    MMP

    去看正解。。
    复杂度一样好不好。。
    每次到达一个点，记录其子树中所有的dp值
    优先向大的一边转移
*/
#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstring>

#define INF 1e8
const int BUF = ;
char Buf[BUF], *buf = Buf;

inline void read (int &now)
{
    for (now = ; !isdigit (*buf); ++ buf);
    for (; isdigit (*buf); now = now *  + *buf - '', ++ buf);
}

#define Max 1100

struct E
{
    E *n;
    int to;
};
E *list[Max], poor[Max << ], *Tail = poor;
int father[Max], dp[Max];

inline int max (int a, int b)
{
    return a > b ? a : b;
}
inline int min (int a, int b)
{
    return a < b ? a : b;
}
void Dp (int now, int Father)
{
    int data[Max], C = ;
    father[now] = Father;
    for (E *e = list[now]; e; e = e->n)
        if (e->to != Father)
        {
            Dp (e->to, now);
            data[++ C] = dp[e->to];
        }
    std :: sort (data + , data +  + C);
    for (int i = ; i <= C; ++ i)
        dp[now] = max (dp[now], data[i] + C - i + );

}
int Answer[Max];

int Main ()
{
    fread (buf, , BUF, stdin); int Maxn = INF;
    int N, M; read (N); register int i; int x;
    for (i = ; i <= N; ++ i)
    {
        read (x);
        ++ Tail, Tail->to = i, Tail->n = list[x], list[x] = Tail;
        ++ Tail, Tail->to = x, Tail->n = list[i], list[i] = Tail;
    }
    for (i = ; i <= N; ++ i)
    {
        Dp (i, );
        Answer[i] = dp[i];
        Maxn = min (Maxn, Answer[i]);
        memset (dp, , sizeof dp);
    }
    printf ("%d\n", Maxn + );
    for (i = ; i <= N; ++ i)
        if (Answer[i] == Maxn) printf ("%d ", i);
    return ;
}
int ZlycerQan = Main ();
int main (int argc, char *argv[]) {;}