【编程测试题】头条校招

题目描述

头条的2017校招开始了!为了这次校招,我们组织了一个规模宏大的出题团队,每个出题人都出了一些有趣的题目,而我们现在想把这些题目组合成若干场考试出来,在选题之前,我们对题目进行了盲审,并定出了每道题的难度系统。一场考试包含3道开放性题目,假设他们的难度从小到大分别为a,b,c,我们希望这3道题能满足下列条件:
a<=b<=c
b-a<=10
c-b<=10
所有出题人一共出了n道开放性题目。现在我们想把这n道题分布到若干场考试中(1场或多场,每道题都必须使用且只能用一次),然而由于上述条件的限制,可能有一些考试没法凑够3道题,因此出题人就需要多出一些适当难度的题目来让每场考试都达到要求,然而我们出题已经出得很累了,你能计算出我们最少还需要再出几道题吗?

输入描述:

输入的第一行包含一个整数n,表示目前已经出好的题目数量。

第二行给出每道题目的难度系数d1,d2,...,dn。 

数据范围

对于30%的数据,1 ≤ n,di ≤ 5;

对于100%的数据,1 ≤ n ≤ 10^5,1 ≤ di ≤ 100。

在样例中,一种可行的方案是添加2个难度分别为20和50的题目,这样可以组合成两场考试:(20 20 23)和(35,40,50)。

输出描述:

输出只包括一行,即所求的答案。

#include<iostream>
#include<vector>
#include <algorithm>
using namespace std;
 
int main()
{
	int n;
	//int *a = new int[n];
	vector<int> a;
	int i, j;
	int out=0;
	cin >> n;
	int aa;
	for(i=0; i<n; ++i)
	{
		cin >> aa;
		a.push_back(aa);
	}
	sort(a.begin(), a.end());
	int num;
	for(i=0; i<n; ++i)//从前往后遍历
	{
		num=1;
		if(i<n-2)
		{
			if(a[i+1]-a[i] <= 10)
			{
				num++;
				if(a[i+2]-a[i+1] <= 10)
				{
					num++;
					i+=2;
				}
				else
				{
					num++;
					out++;
					i++;
				}
			}
			else if(a[i+1]-a[i] <= 20)
			{
				num+=2;
				out++;
				i++;
			}
			else
			{
				num+=2;
				out+=2;	
			}
		}
		else if(i==n-2)//如果只剩下两个元素
		{
			if(a[i+1]-a[i]<=20)
			{
				num+=2;
				out++;
				i++;
			}
			else
			{
				num+=2;
				out+=4;
				i++;
			}
		}
	}
	if(num!=3)//如果只剩下一个元素
		out += (3-num);
	cout << out << endl;
 
}

贪婪算法,排序后有符合条件的就计数,不符合的再看加几个符合。

猜你喜欢

转载自blog.csdn.net/HelloZEX/article/details/81703813