题解-今日头条2018校招后端方向(第二批)

原题链接

链接:https://www.nowcoder.net/questionTerminal/429c2c5a984540d5ab7b6fa6f0aaa8b5

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


----------------------------------分界线---------------------------------

分析:

对同色串珠的位置进行判断即可
那么问题就是用什么方式保存珠串

介绍两个办法 第一个比较容易想到  第二个是基于第一个的优化

方法一:(容易想到)
采取多个向量 vector 保存不同颜色珠串的位置信息
c种颜色 则有  vector<int> vec[c]
vec[i]中保存颜色为 i 的所有珠串的位置,因为是按顺序读取,所以vec[i]中的位置信息也是有序的
另外,手串是一个环,要判断连续 m 个珠串,则可能出现 末尾与头连接的情况
那么,判断连续 m 个是否出现相同颜色,只需判断 同颜色相邻两个的位置是否小于 m
考虑环形情况,还需考虑同颜色第一个出现的位置和最后一个出现的位置差

方法二:(优化空间复杂度)
其实不必记录每个位置信息,只需记录每种颜色上一个出现的位置即可
另外考虑环状绕一圈回来,还需记录每种颜色第一次出现的位置
这样 空间复杂度就优化到了 O(c)  c 为颜色的种数,只需要有限的一点空间即可 


----------------------------------分界线---------------

方法一 (容易想到)

vector 示例

输入

5 2 3
3 1 2 3
0
2 2 3
1 2
1 3
此时
vector[1]: 0 3 4    //1号颜色出现的位置为 0号 3号 4号 珠
vector[2]: 0 2 3
vector[3]: 0 2 4 

代码:


#include<stdio.h>
#include<vector>
#include<math.h>
#include<algorithm>
using namespace std;
int main(){
    int ans=0,n,m,c,times,color,i,j;
    scanf("%d %d %d",&n,&m,&c);
    vector<int>vec[c+1];
    //按颜色分类记录出现的位置 
    for(i=0;i<n;i++){
        scanf("%d",×);
        for(int j=0;j<times;j++){
            scanf("%d",&color);
            vec[color].push_back(i);
        }
    }
    //判断连续m个是否出现重复
	//只需对每个颜色判断同颜色相邻的位置差是否小于m 
    for(i=1;i<=c;i++){
    	//先判断环形头尾连接,即最后一个和第一个出现的位置差 
		if(vec[i][0]+n-vec[i][vec[i].size()-1]<m){
			ans++; 
			continue;
		}
		//判断每个相邻位置差 
        for(j=0;j<vec[i].size()-1;j++){
            if(vec[i][j+1]-vec[i][j]<m){
                ans++;
                break;
            }
        }
    }
    printf("%d",ans);
    return 0;
}



----------------------------------分界线---------------

方法二 (优化空间复杂度)


#include<stdio.h>
#include<cstring>
#include<math.h>
#include<algorithm>
using namespace std;
int main(){
    int ans=0,n,m,c,times,color,i,j;
    scanf("%d %d %d",&n,&m,&c);
    int last_pos[c+1];  //每种颜色上一次出现的位置 
    int first_pos[c+1]; //每种颜色第一次出现的位置 
    bool ok[c+1];       //每种颜色是否满足题意 
    memset(first_pos,0,sizeof(first_pos));
    memset(last_pos,0,sizeof(last_pos));
    memset(ok,1,sizeof(ok));
    
    for(i=1;i<=n;i++){
        scanf("%d",×);
        for(int j=0;j<times;j++){
            scanf("%d",&color);
            if(ok[color]){
            	if(!first_pos[color])first_pos[color]=i;	//记录第一次出现的位置 
            	else if(i-last_pos[color]<m)ok[color]=false; //判断非环形情况 
				last_pos[color]=i;		//更新上一次出现的位置 
            }
        }
    }
    for(i=1;i<=c;i++){
    	if(ok[i]&&first_pos[i]+n-last_pos[i]<m)ans++; //用头尾判断环形情况 
    	if(!ok[i])ans++;
    }
    printf("%d",ans);
    return 0;
}


猜你喜欢

转载自blog.csdn.net/bobbymly/article/details/79289575