CSU 2124智慧树(建图+BFS)

CSU 2124智慧树(建图+BFS)

Description

图书馆门前的那两棵树象征着爱与智慧,吸引着很多毕业学子和情侣前来拍照。

某天Wells一个人在智慧树下无聊的数树叶,却惊奇的发现其实智慧树其实也有很重的浪漫气息

Wells发现树上的每个结点都有自己相应的浪漫值,现在她想知道每一层的最大结点浪漫值是多少?

Input

多组数据

第一行一个正整数n,表述n个结点

接着n个正整数,a1,a2,a3,…,a_n表示n个结点的浪漫值

后面一行n-1个正整数,(第i个数字v表示,结点i+1的父亲是v,保证v

Output

根结点为第一层,从根结点开始输出每一层的最大值,用空格隔开,末尾换行。

Sample Input

6

10 6 8 7 3 2

1 2 1 4 4

Sample Output

10 7 8

题意

  给出n个节点以及他们的权值,然后再告诉n个节点之间的关系,求树的每一层的最大值。

解题思路

  由于数据范围是1e5,可以建图,然后直接用BFS(爆发式)算法遍历,最大值用一个数组存即可。至于建图,我用前向星处理了一下。

代码

#include
#include
#include
#include
#include
using namespace std;
const int maxn = 1e5+5;

int arr[maxn],maxx[maxn];
struct node
{
    int x,step;
    node() {}
    node(int a,int b):x(a),step(b) {}
};
struct edge
{
    int to,next;
} edges[maxn*2];
int head[maxn],cnt,n;
int vis[maxn];

void init()
{
    cnt=1;
    memset(head,-1,sizeof(head));
}
void addedges(int u,int v)
{
    edges[cnt].to=v;
    edges[cnt].next=head[u];
    head[u]=cnt++;
}
void bfs()
{
    memset(vis,0,sizeof(vis));
    queue q;
    q.push(node(1,1));
    vis[1]=1;
    maxx[1]=arr[1];
    while(!q.empty())
    {
        node x=q.front();
        q.pop();
        for(int i=head[x.x]; ~i; i=edges[i].next)
        {
            if(vis[edges[i].to]) continue;
            vis[edges[i].to]=1;
            maxx[x.step+1]=max(arr[edges[i].to],maxx[x.step+1]);
            q.push(node(edges[i].to,x.step+1));
        }
    }
}
int main()
{
    int n;
    while(~scanf("%d",&n))
    {
        for(int i=1; i<=n; i++)
            scanf("%d",&arr[i]);
        int x;
        init();
        for(int i=1;iscanf("%d",&x);
            addedges(x,i+1);
        }
        memset(maxx,-1,sizeof(maxx));
        bfs();
        for(int i=1;maxx[i]!=-1;i++)
            printf("%s%d",i==1?"":" ",maxx[i]);
        printf("\n");
    }
}

补一份vector建图

#include
#include
#include
#include
#include
using namespace std;
const int maxn = 1e5+5;

int arr[maxn],maxx[maxn];
struct node
{
    int x,step;
    node() {}
    node(int a,int b):x(a),step(b) {}
};
int n,vis[maxn];
vector v[maxn];

void bfs()
{
    memset(vis,0,sizeof(vis));
    queue q;
    q.push(node(1,1));
    vis[1]=1;
    maxx[1]=arr[1];
    while(!q.empty())
    {
        node x=q.front();
        q.pop();
        for(int i=0; i

你可能感兴趣的:(数据结构-链式前向星,图论-BFS)