笔试刷题-头条2018-06-02

题目介绍：

/**
作为一个手串艺人，有金主向你订购了一条包含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

输出例子1:
2

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

思路简介：
colorPosTable[c]表示第c种颜色出现的位置（按照顺时针录入）

colorSet记录颜色的种类

遍历colorSet的各种颜色

然后根据colorPosTable[c]中看成一个环，是否存在pos1 pos2 pos3其中pos1 pos2 pos3是列表中相邻（顺时针数）的三个下标然后看

int curLeft = (cur-1+sz)%sz;

int distLeft = abs(colorPosTable[color][cur]-colorPosTable[color][curLeft]);

（环中两点距离计算短的那段） distLeft = min(distLeft, N-distLeft);

若pos2向左 向右距离 pos1 pos3距离都小于M那么pos2就是一个符合题目要求的位置，计数器加一

代码如下：

#include<stdio.h>
#include<iostream>
#include<set>
#include<vector>

#define MAX_N 10005
#define MAX_M 1005
#define MAX_C 55

using namespace std;

//colorPosTable[c]表示第c种颜色出现的位置,若colorPosTable[c]=0表示无色阻断（先按题目无色不阻断尝试）
vector<int> colorPosTable[MAX_C];
set<int> colorSet;

int main(){
    int N, M, C;
    scanf("%d%d%d", &N, &M, &C);
    for(int n=0; n<N; n++){
        int colorNum;
        scanf("%d", &colorNum);
        for(int c=0; c<colorNum; c++){
            int color;
            scanf("%d", &color);
            colorSet.insert(color);
            colorPosTable[color].push_back(n);
        }
    }
    int cnt=0;
    set<int>::iterator it;
    for(it=colorSet.begin(); it!=colorSet.end(); it++){
        int color = (*it);
        int sz = colorPosTable[color].size();
        if(sz==1){
            continue;
        }
        for(int cur=0; cur<sz; cur++){
            int curLeft = (cur-1+sz)%sz;
            int curRight = (cur+1)%sz;
            int distLeft = abs(colorPosTable[color][cur]-colorPosTable[color][curLeft]);
            distLeft = min(distLeft, N-distLeft);
            int distRight = abs(colorPosTable[color][cur]-colorPosTable[color][curRight]);
            distRight = min(distRight, N-distRight);
            if(distLeft<M || distRight<M){
                cnt++;
                break;
            }
        }
    }
    printf("%d", cnt);
    return 0;
}