码迷,mamicode.com
首页 > 其他好文 > 详细

codeforces902B. Coloring a Tree

时间:2018-08-14 22:59:40      阅读:197      评论:0      收藏:0      [点我收藏+]

标签:algorithm   string   前向星   ring   pair   blank   ace   ack   ems   

B. Coloring a Tree

题目链接:

https://codeforces.com/contest/902/problem/B

题意:给你一颗树,原先是没有颜色的,需要你给树填色成指定的样子,每次填色的话,子树会和根节点变成同一种颜色,问需要多少次填色

题解:前向星建树,从每个父节点便利,如果父节点的颜色和子节点不一样就ans++吗,最后记得把第一次操作(根节点涂色的次数 给记上

代码如下:

技术分享图片
#include <map>
#include <set>
#include <cmath>
#include <ctime>
#include <stack>
#include <queue>
#include <cstdio>
#include <cctype>
#include <bitset>
#include <string>
#include <vector>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <functional>
#define fuck(x) cout<<"["<<x<<"]";
#define FIN freopen("input.txt","r",stdin);
#define FOUT freopen("output.txt","w+",stdout);
//#pragma comment(linker, "/STACK:102400000,102400000")
using namespace std;
typedef long long LL;
typedef pair<int, int> PII;
const int maxn = 3e5+5;
const int INF = 0x3f3f3f3f;
const int MOD = 1e9+7;
int cnt;
int head[maxn],nxt[maxn],to[maxn];
int color[maxn];
int ans;
void add(int x,int y){
    to[cnt]=y;
    nxt[cnt]=head[x];
    head[x]=cnt++;
}

int main(){
#ifndef ONLINE_JUDGE
    FIN
#endif
    int n,x;
    cnt=1;
    memset(head,-1,sizeof(head));
    scanf("%d",&n);
    for(int i=2;i<=n;i++){
        scanf("%d",&x);
        add(x,i);
    }//前向星连边
    for(int i=1;i<=n;i++){
        scanf("%d",&color[i]);
    }
    for(int i=1;i<=n;i++){
        for(int j=head[i];j!=-1;j=nxt[j]){
            //如果父节点和子树的节点不一样就操作一次
            if(color[i]!=color[to[j]]) ans++;
        }
    }
    cout<<ans+1<<endl; //根节点的第一次操作算进去

}
View Code

 

codeforces902B. Coloring a Tree

标签:algorithm   string   前向星   ring   pair   blank   ace   ack   ems   

原文地址:https://www.cnblogs.com/buerdepepeqi/p/9478176.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!