编程之美 彩色的树

题目1 : 彩色的树

时间限制:2000ms

单点时限:1000ms

内存限制:256MB

描述

给定一棵n个节点的树，节点编号为1, 2, …, n。树中有n - 1条边，任意两个节点间恰好有一条路径。这是一棵彩色的树，每个节点恰好可以染一种颜色。初始时，所有节点的颜色都为0。现在需要实现两种操作:

1. 改变节点x的颜色为y；

2. 询问整棵树被划分成了多少棵颜色相同的子树。即每棵子树内的节点颜色都相同，而相邻子树的颜色不同。

输入

第一行一个整数T，表示数据组数，以下是T组数据。

每组数据第一行是n，表示树的节点个数。接下来n - 1行每行两个数i和j，表示节点i和j间有一条边。接下来是一个数q，表示操作数。之后q行，每行表示以下两种操作之一：

1. 若为"1"，则询问划分的子树个数。

2. 若为"2 x y"，则将节点x的颜色改为y。

输出

每组数据的第一行为"Case #X:"，X为测试数据编号，从1开始。

接下来的每一行，对于每一个询问，输出一个整数，为划分成的子树个数。

数据范围

1 ≤ T ≤ 20

0 ≤ y ≤ 100000

小数据

1 ≤ n, q ≤ 5000

大数据

1 ≤ n, q ≤ 100000

样例输入

231 22 3312 2 1151 22 32 42 5412 2 12 3 21

样例输出

Case #1:13Case #2:1
5

#include<iostream>#include <vector>#include <fstream>using namespace std;const int maxVex = 100010;vector<int> parent[maxVex];vector<int> child[maxVex];int color[maxVex];bool visited[maxVex];int num;//tree node number//ifstream in("/Users/urey/data/input");void create() {    int x,y;    cin>>num;    memset(parent,0,sizeof(parent));    memset(child,0,sizeof(child));    memset(color,0,sizeof(color));    for(int i=1; i<=num-1; i++) {        cin>>x>>y;        child[x].push_back(y);//x's child is y        parent[y].push_back(x);//y's parent is x    }}void dfs(int i) {    visited[i]=true;    for(auto j=child[i].begin();j!=child[i].end();j=next(j)) {        if(*j<=num && *j>=1 && color[*j]==color[i] ) {            if(!visited[*j]) dfs(*j);        }    }    for(auto j=parent[i].begin();j!=parent[i].end();j=next(j)) {        if(*j<=num && *j>=1 && color[*j]==color[i] ) {            if(!visited[*j]) dfs(*j);        }    }}int dfsTraverse() {    int ans=0;    memset(visited,false,sizeof(visited));    for(int i=1; i<=num; i++) {        if(!visited[i]) {            dfs(i);            ans++;        }    }    return ans;}int main() {    int T,q,x,y,ops;    cin>>T;    for(int i=0; i<T; i++) {        create();        cin>>q;        cout<<"Case #"<<i+1<<":"<<endl;        for(int j=0; j<q; j++) {            cin>>ops;            if(ops == 1) {                cout<<dfsTraverse()<<endl;            }            if(ops == 2) {                cin>>x>>y;                color[x]=y;            }        }    }    return 0;}