今日头条2018校招算法方向--手串问题

题目:

作为一个手串艺人,有金主向你订购了一条包含n个杂色串珠的手串——每个串珠要么无色,要么涂了若干种颜色。为了使手串的色彩看起来不那么单调,金主要求,手串上的任意一种颜色(不包含无色),在任意连续的m个串珠里至多出现一次(注意这里手串是一个环形)。手串上的颜色一共有c种。现在按顺时针序告诉你n个串珠的手串上,每个串珠用所包含的颜色分别有哪些。请你判断该手串上有多少种颜色不符合要求。即询问有多少种颜色在任意连续m个串珠中出现了至少两次。
 

输入描述:

第一行输入n,m,c三个数,用空格隔开。(1 <= n <= 10000, 1 <= m <= 1000, 1 <= c <= 50) 接下来n行每行的第一个数num_i(0 <= num_i <= c)表示第i颗珠子有多少种颜色。接下来依次读入num_i个数字,每个数字x表示第i颗柱子上包含第x种颜色(1 <= x <= c)

输出描述:

一个非负整数,表示该手链上有多少种颜色不符需求。

示例1

输入

5 2 3
3 1 2 3
0
2 2 3
1 2
1 3

输出

2

说明

第一种颜色出现在第1颗串珠,与规则无冲突。
第二种颜色分别出现在第 1,3,4颗串珠,第3颗与第4颗串珠相邻,所以不合要求。
第三种颜色分别出现在第1,3,5颗串珠,第5颗串珠的下一个是第1颗,所以不合要求。
总计有2种颜色的分布是有问题的。 
这里第2颗串珠是透明的。

解题思路:

假设所有颜色初始时都是合格的。

遍历所有珠子。对于每一个颜色,若是第一次出现,记录下初始位置和当前的位置。若某一颜色不是第一次出现,则判断该颜色当前位置和上一次出现的位置之间的距离是否小于m。若小于m,则这个颜色标记为不合格,并在接下来的循环中不再考虑这个颜色,否则,说明该颜色合格,更新该颜色当前的位置为当前珠子。

因为手串为环状,所以在循环最后,要继续判断合格的颜色中,距离第一次出现的位置的距离是否小于m。

若颜色为无色,不计入不合要求的颜色中

#include<iostream>
using namespace std;
int main(){
    int n,m,c;
    int count;
    scanf("%d %d %d",&n,&m,&c);
    int color_init_add[c+1];//颜色初始位置
    int color_add[c+1];//颜色位置
    int flag[c+1];//颜色是否合格
    for(int i=1;i<=c;i++){
        color_init_add[i] = -1;
        color_add[i] = -1;
        flag[i] = 1;//初始时颜色都合要求
    }
    int k;
    for(int i=1;i<=n;i++){
        cin>>count;
        for(int j=0;j<count;j++)
        {
            cin>>k;
            if(flag[k]==1){
                if(color_init_add[k] == -1) color_init_add[k] = i;
                if(color_add[k] == -1) color_add[k] = i;
                else{
                    if(i-color_add[k]<m)
                        flag[k] = -1;
                    else
                        color_add[k] = i;
                }
            }
        }
    }
    for(int j=1;j<=c;j++){
        if(flag[j] == 1){
            if(n-color_add[j]+color_init_add[j]<m)
                flag[j] = -1;
        }
    }
    int sum = 0;
    for(int i=1;i<=c;i++){
        if(flag[i] == -1)
            sum++;
    }
    cout<<sum;
    return 0;
}

猜你喜欢

转载自blog.csdn.net/dlwsd_/article/details/82425166