Hulu面試題解答——N位數去除K個數字(解法錯誤sorry)

給定一個N位數,例如12345,從裏面去掉k個數字,得到一個N-k位的數,例如去掉2,4,得到135,去掉1,5,得到234。設計算法,求出所有得到的N-k位數裏面最小的那一個。

寫的代碼如下,思路是通過堆排序得到N位數裏邊最大的前K個數,然後按照原數字的順序去除得到的最大的K個數。感覺寫的很亂,可能還有些小問題,魯棒性應該很差,努力鍛鍊。。努力提高!

typedef unsigned int uint;
//Heap adjust function
void HeapAdjust(uint *value, uint start, uint end)
{
	for( int i=start; 2*i<=end; )
	{
		uint index = 2*i;
		if(index+1<=end && value[index+1] > value[index])
			index +=1;
		if(value[index] > value[i])
		{
			uint temp = value[i];
			value[i] = value[index];
			value[index] = temp;
		}
			i = index;
	}
}
//Heap creation function
void CreateHeap(uint *value, uint start, uint end)
{
	uint middle = (start+end)/2;
	for( uint i = middle;i>=start;i--)
	{
		HeapAdjust(value, i, end);
	}
}
//Kick K numbers from N
uint KickK(uint N, uint K)
{
	if(N>4294967295)
		return 0;
	uint i,tempN=N;
	uint length=0;
	uint value[10],outValue[10];
	bool flag;
	i = 1;
	//get the numbers of N 
	while(tempN != 0)
	{
		value[i]=tempN%10;
		outValue[i]=value[i];
		tempN=tempN/10;
		i++;
	}
	length = i-1;
	//check the condition
	if(K >= length)
		return 0;
	//Get the K biggest numbers, and store then in the last K positions in array value
	CreateHeap(value,1,length);
	for(i=0;i<K;i++)
	{
		uint temp = value[length-i];
		value[length-i] = value[1];
		value[1] = temp;
		CreateHeap(value,1,length-i-1);
	}
	//output the remained numbers ,despite the K biggest numbers
	for(i=length;i>=1;i--)
	{
		flag = true;
		for(uint j=length+1-K;j<=length; j++)
		{
			if(outValue[i] == value[j])
				flag = false;
		}
		if(flag)
			printf("%d", outValue[i]);
	}
	return 1;
}

int main()
{
	uint N=61829;
	uint K=3;
	KickK(N,K);
}


發佈了63 篇原創文章 · 獲贊 18 · 訪問量 15萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章