互联网公司面试题之十二

问题：五连击数组[BAIDU]：在任意一个数组当中，若你能找出有五个以及五个以上的连续元素时，我们就称它为五连击数组。但是这种数组并不是视场会有，你的任务就是通过增加最少的元素，使一个数组成为一个五连击数组。

输入要求：每个测试文件包括多个测试案例，每个测试案例包括两行。第一行代表输入的元素个数N，其中0<N<=1000。第二行则包含有N个非负整数，代表数组的元素。

输出要求：对于每一个测试案例输出一行，代表最少添加的元素数，使得数组成为一个五连击数组。


答：代码实现如下：

/* Author:YuHuang  Date:2012-01-24 */

#include <cstdlib>
#include <iostream>
#include <algorithm>

using namespace std;

bool compare(unsigned long a,unsigned long b){
	return a<b;
}

int main()
{
	int n,i,j,c,m;
	unsigned long a[1000],w;
	for(;cin>>n;){
		for(i=0;i<n;++i) cin>>a[i];
		sort(a,a+n,compare);
		for(m=5,i=0;i<n;++i){
			w=a[i];
			for(c=5,j=i;w<a[i]+5&&j<n;){
				if(a[j]==w) {--c;++j;} 
				else ++w;
			}
			if(c<m) m=c;
		}
		cout<<m<<endl;
	}
	return 0;
}

代码运行结果：