字节跳动2018校招后端方向—手串问题(C++)

时间限制:C/C++ 1秒,其他语言2秒

空间限制:C/C++ 64M,其他语言128M

作为一个手串艺人,有金主向你订购了一条包含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颗串珠是透明的。

#include
#include
using namespace std;
/*
算法:
1、求出颜色i 出现的位置,保存在vectorpos中
2、判断相邻位置的差是否小于m,若小于,表示颜色i为不合格颜色(注意要判断第一个位置和最后一个位置的差)
*/
int main()
{
     
	int n, m, k;//珠子数量,连续数量,颜色数
	cin >> n >> m >> k;
	vector<vector<int>>chuanZhuNum(n);
	for (int i = 0; i < n; i++)
	{
     
		int p;
		cin >> p;//每颗珠子颜色种数
		for (int j = 0; j < p; j++)
		{
     
			int h;
			cin >> h;
			chuanZhuNum[i].push_back(h);//具体颜色
		}
	}


	int res = 0;//保存结果
	//每一次判断一种颜色是否合格
	for (int i = 1; i <= k; i++)
	{
     
		//储存颜色i出现在哪些珠子上(pos为0到n-1)
		vector<int>pos;
		for (int g = 0; g < chuanZhuNum.size(); g++)
		{
     
			for (int h = 0; h < chuanZhuNum[g].size(); h++)
			{
     
				if (i == chuanZhuNum[g][h])
				{
     
					pos.push_back(g);//g:代表第g颗珠子
					break;//只要在第g颗珠子上找到颜色i,就不需要在第g行继续找了
				}
			}
		}

		//判断连续m个珠子颜色i是否出现两次以上
		//只要任意两个位置只差小于m,则颜色i不满足要求
		if (pos[pos.size() - 1] -pos[0]<m)//判断最后一个位置与第一个位置是否满足要求
			res++;
		else
		{
     
			for (int g = 0; g < pos.size() - 1; g++)
			{
     
				if (pos[g + 1] - pos[g] < m)
				{
     
					res++;
					break;//只要有一组不满足,则颜色i不合格
				}
			}
		}
	}	
	cout << res << endl;
	system("pause");
	return 0;
}

你可能感兴趣的:(企业面试题,c++)