今日头条2017秋招笔试编程题

手串

题目大意：

        有一条环形手串，手串由若干个宝石组成，每个宝石有若干种颜色或无色。我们规定如果手串上长为m的一段宝石中出现了相同的颜色，则该颜色不符合条件。给你手串的长度n ( 1 <= n <= 10000 )、宝石段的长度m ( 1<= m <=n )、手串中的总颜色数c( 1 <= c <=50 )以及每个宝石上具体的颜色种类，请你求出不合法的颜色的总数。

输入描述：

第一行输入n,m,c；

随后n行每行先输入一个数k（0 <= k <= c）,代表该颗宝石拥有的颜色的总数（0代表无色），接下来输入k个数字x[i] ( 1 <= x[i] <= c )，为对应颜色的序号；

输出描述：

输出不合法的颜色的总数；

输入样例：

5 2 3

3 1 2 3

0

2 2 3

1 2

1 3

样例输出：

2

样例解析：

        对于长为2的宝石段，第三颗宝石和第四颗宝石都有颜色2，第五颗宝石和第一颗宝石都有颜色1，颜色1和2不合法，输出不合法颜色种数为2。

解题思路：

      手串是一个环形，处理环只要把手串的数据复制一遍连接在原始数据后面（如123变成123123），就可以将其转化为线形。接下来就是遍历一般数据，统计一下长为m的每一段宝石上各种颜色出现次数的总数，如果有颜色出现次数大于一次，则将该颜色标记为不合法，最后统计不合法的颜色的总数。统计每一段宝石上各种颜色的总数时，只需要用他前面那一段宝石的总数据减去前面那一段的第一颗宝石的颜色数据再加上当前宝石的数据就可以了。由于颜色数量足够小，所以可以认为该算法的复杂度为O（n）;

AC代码：

#include<iostream>
#include<queue>
#include<vector>
#include<map>
#include<string.h>
#include<math.h>
using namespace std;

int num[20001][51],sum[20001][51],mark[51];

int main()
{
	int n,m,c,k,i,j,p,s=0;
	cin>>n>>m>>c;
	for(i=0;i<n;i++)
	{
		cin>>k;
		while(k-->0)
		{
			cin>>p;
			num[i][p]=1;
		}
	}
	for(i=n;i<n+m;i++)
	{
		for(j=1;j<=c;j++)
		num[i][j]=num[i-n][j];
	}
	for(j=1;j<=c;j++)
	sum[0][j]=num[0][j];
	for(i=1;i<m;i++)
	{
		for(j=1;j<=c;j++)
		{
			sum[i][j]=num[i][j]+sum[i-1][j];
			//cout<<sum[i][j]<<" ";
		}
		//cout<<endl;
	}
	for(i=m;i<m+n;i++)
	{
		for(j=1;j<=c;j++)
		{
			sum[i][j]=num[i][j]+sum[i-1][j]-num[i-m][j];
			//cout<<sum[i][j]<<" ";
		}
		//cout<<endl;
	}
	for(i=m-1;i<m+n;i++)
	{
		for(j=1;j<=c;j++)
		if(sum[i][j]>=2)
		mark[j]=1;
	}
	for(i=1;i<=c;i++)
	s+=mark[i];
	cout<<s<<endl;
	return 0;
}

用户喜好

题目大意：

今日头条有很多用户，头条统计了每个用户对某种文章的喜好度，给你n个用户（编号为1-n）对某种文章的喜好度x[i]，让你查询编号在l到r之间的用户中喜好度为k的用户的人数。

输入描述：

第一行输入一个数n（1 <= n <= 300000），代表用户的人数；

接下来一行n个整数（没有具体范围），表示每个用户的喜好度；

接下来一行有一个整数p（1 <= p <= 300000），代表查询的数量；

接下来p行每行有三个整数 l  r  k  ( 1<=l<=r<=n )，表示查询编号在l-r（包含l和r）之间的用户中喜好度为k的用户的人数；

输出描述：

对于每个查询输出一行，为该查询中符合条件的人数。

输入样例：

5

1 2 3 3 4

3

1 2 1

2 4 4

3 5 3

样例输出：

1

0

2

样例解析：

         对于第1个查询，在第1个用户到第2个用户之间有1个用户的喜好度为1，输出1；

         对于第2个查询，在第2个用户到第4个用户之间有0个用户的喜好度为4，输出0；

         对于第3个查询，在第3个用户到第5个用户之间有2个用户的喜好度为3，输出2；

解题思路：

       用一个结构体保存每个用户的编号和喜好度，然后对用户按喜好度升序排序，喜好度相同的按序号升序排序，然后用一个map记录每一个喜好度开头的第一个用户在排序后序列中的位置，对于每一次查询，直接找到第一个喜好度为k的用户在序列中的位置，然后看用户的序号是不是在l和r之间，如果喜好度不为k或者序号大于了r，查询结束，输出结果。

AC代码：

#include<iostream>
#include<map>
#include<algorithm>
using namespace std;

struct num{
	int w;
	int d;
};

num a[300005];

bool cmp(num p,num q)
{
	if(p.d==q.d)
	return p.w<q.w;
	else
	return p.d<q.d;
}

int main()
{
	int n,i,l,r,k,s,T;
	map<int,int> Map;
	cin>>n;
	for(i=0;i<n;i++)
	{
		cin>>k;
		a[i].w=i;
		a[i].d=k;
	}
	sort(a,a+n,cmp); 
	Map[a[0].d]=0;
	for(i=1;i<n;i++)
	if(a[i].d!=a[i-1].d)
	{
		Map[a[i].d]=i;
	}
	cin>>T;
	while(T-->0)
	{
		cin>>l>>r>>k;
		l-=1;
		r-=1; 
		s=0;
		for(i=Map[k];i<n;i++)
		{
			if(a[i].d!=k || a[i].w>r)
			break;
			if(a[i].w>=l && a[i].w<=r)
			s++;
		}
		cout<<s<<endl;
	}
	return 0;
}