hihocoder 微软编程之美2015 初赛 第一场 (树算法 + 暴力思想 + 搜索思想)

题目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

样例输入
2

3

1 2

2 3

3

1

2 2 1

1

5

1 2

2 3

2 4

2 5

4

1

2 2 1

2 3 2

1

样例输出
Case #1:

1

3

Case #2:

1

5



分析:主要题意就是输出一棵树中连通分量的个数,但此题目里的连通分量和平时的连通分量不太一样,就是每个连通分量内的
节点的颜色必须是一样的,否则就要划分到不同的连通分量上。常规的思路,每次在修改完某些节点后,但我们想去计算这棵树
中有多少子树(即连通分量),完全可以用dfs搜索,但是这道题目中如果询问次数很多,必然超时!
所以需要修改思路。
假设我们当前修改节点x的颜色,将其颜色修改为y。
在我们修改后,其对应的颜色可能会发生改变,也可能不会。子树的数目可能增加,也可能减少,还有可能不变。
回到节点x,我们修改节点x的颜色后,我们访问所有与x节点有直接相邻关系的节点xx

假设:原来x的颜色为ori, 修改后的颜色为cur,将会产生下面四种情况:( f[]数组保存所有节点的颜色信息 )
ans:表示x节点颜色修改前的子树的个数:接下来计算x节点颜色的修改对子树数目的影响
ori==f[xx] && cur==f[xx]
ans不变
ori==f[xx] && cur!=f[xx] ans++;
ori!=f[xx] && cur==f[xx] ans--;
ori!=f[xx] && cur!=f[xx]
ans不变

代码:(本以为此算法同样可以过大数据,但还是TLE了)
#include <stdio.h>

#include <string.h>

#include <stdlib.h>

#include <ctype.h>

#include <math.h>

#include <algorithm>

#define N 5010





using namespace std;



int map[N][N];

int fa[N];

bool vis[N];

int n, q;



int main()

{

    int t;

    scanf("%d", &t);

    int i, j, k;

    int cnt=1;



    while(t--)

    {

        scanf("%d", &n);

        memset(fa, 0, sizeof(fa));

        memset(map, 0, sizeof(map));



        int u, v;

        for(i=0; i<n-1; i++)

        {

            scanf("%d %d", &u, &v);

            map[u][v]=1;

            map[v][u]=1;

        }





        scanf("%d", &q);

        printf("Case #%d:\n", cnt++ );

        int ans=1;

        while(q--)

        {

            int dd;

            int x, y;

            scanf("%d", &dd);

            if(dd==1)

            {

                printf("%d\n", ans );

            }

            else

            {

                scanf("%d %d", &x, &y);

                int ori=fa[x];

                fa[x]=y;

                int cur=y;

                for(i=1; i<=n; i++)

                {

                    if(map[x][i]==1 && i!=x )

                    {

                        if(ori==fa[i] && cur!=fa[i] )

                        {

                            ans++;

                        }

                        else if(ori!=fa[i] && cur!=fa[i])

                        {

                            ans=ans+0;

                        }

                        else if(ori==fa[i] && cur==fa[i])

                        {

                            ans+=0;

                        }

                        else if(ori!=fa[i] && cur==fa[i])

                        {

                            ans--;

                        }

                    }

                }

            }

        }

    }

    return 0;

}

 

题目3 : 质数相关

时间限制: 2000ms
单点时限: 1000ms
内存限制: 256MB

描述

两个数a和 b (a<b)被称为质数相关,是指a × p = b,这里p是一个质数。一个集合S被称为质数相关,是指S中存在两个质数相关的数,否则称S为质数无关。如{2, 8, 17}质数无关,但{2, 8, 16}, {3, 6}质数相关。现在给定一个集合S,问S的所有质数无关子集中,最大的子集的大小。

输入

第一行为一个数T,为数据组数。之后每组数据包含两行。

第一行为N,为集合S的大小。第二行为N个整数,表示集合内的数。

输出

对于每组数据输出一行,形如"Case #X: Y"。X为数据编号,从1开始,Y为最大的子集的大小。

数据范围

1 ≤ T ≤ 20

集合S内的数两两不同且范围在1到500000之间。

小数据

1 ≤ N ≤ 15

大数据

1 ≤ N ≤ 1000

样例输入
3

5

2 4 8 16 32

5

2 3 4 6 9

3

1 2 3

样例输出
Case #1: 3

Case #2: 3

Case #3: 2


此题目:小数据的情况下,直接暴力就可以了,大数据的算法没想出来!
代码:
#include <stdio.h>

#include <string.h>

#include <stdlib.h>

#include <ctype.h>

#include <math.h>

#include <vector>

#include <algorithm>

#define N 100000+10





using namespace std;

int f[500002];

void sushu()

{

    memset(f, 0, sizeof(f));

    int i=2;

    f[1]=1;

    f[0]=1;

    int dd=sqrt(500000+0.5);

    while(i<=dd)

    {

        for(int j=i*2; j<=500000; j+=i)

        {

            f[j]=1;

        }

        i++;

        while(f[i]==1)

            i++;

    }

}



int a[1010];

int b[1010], e=0;



int main()

{

    int t;

    scanf("%d", &t);

    sushu();

    int n;

    int i, j, k;



    int dd=1;

    while(t--)

    {

        scanf("%d", &n);

        for(i=0; i<n; i++)

        {

            scanf("%d", &a[i]);

        }

        int len=0;

        bool flag;

        int cnt;

        for(i=0; i<n; i++)

        {



            memset(b, 0, sizeof(b));

            e=0;

            b[e++]=a[i];

            for(j=i+1; j<n; j++)

            {

                flag=true;

                for(k=0; k<e; k++)

                {

                    if(a[j]%b[k]==0 && f[a[j]/b[k]]==0 )

                    {

                        flag=false;

                        break;

                    }

                }

                if(flag==true)

                    b[e++]=a[j];

            }

            if(e > len )

                len = e;

        }

        printf("Case #%d: %d\n", dd++, len );

    }

    return 0;

}

 

 
           
 
      

你可能感兴趣的:(code)