1. 程式人生 > >Hihocoder 1514

Hihocoder 1514

題目描述:

#1514 : 偶像的條件

時間限制:10000ms 單點時限:1000ms 記憶體限制:256MB

描述

小Hi的學校正面臨著廢校的大危機。面對學校的危機,小Hi同學們決定從ABC三個班中各挑出一名同學成為偶像。  

成為偶像團體的條件之一,就是3名團員之間的身高差越小越好。  

已知ABC三個班同學的身高分別是A1..AN, B1..BM 和 C1..CL。請你從中選出3名同學Ai, Bj, Ck使得D=|Ai-Bj|+|Bj-Ck|+|Ck-Ai|最小。

輸入

第一行包含3個整數,N, M和L。  

第二行包含N個整數,A1, A2, ... AN。(1 <= Ai

 <= 100000000)

第三行包含M個整數,B1, B2, ... BM。(1 <= Bi <= 100000000)

第四行包含L個整數,C1, C2, ... CL。(1 <= Ci <= 100000000)

對於30%的資料, 1 <= N, M, L <= 100  

對於60%的資料,1 <= N, M, L <= 1000  

對於100%的資料,1 <= N, M, L <= 100000

輸出

輸出最小的D。

樣例輸入
3 3 3  
170 180 190  
195 185 175  
180 160 200
樣例輸出
10
解題思路:

這道題目只要理解了題目所要求的數學表達,查詢時結合二分查詢就行了。

由於|A-B|即為數軸上A到B的距離,由目標變數D的數學表示式可知,D為數軸上三點A,B,C,則要使得兩兩之間的距離和最小,等價於三者中最小值與最大值差值的二倍(簡單在數軸上畫一畫就能看出來),由題意可知,A,B,C三點分別為三個班某個同學的身高,根據上邊的理解求解即可:

先將三個班級的同學身高分別排序,假設最優解所取的三個點中最小值為A班的同學,即依次從小到大列舉A班級同學的身高,然後分別在B,C兩個班級尋找第一個不小於此身高的同學(因為假設A班同學是最小值),A與B,C兩同學的差值最大值即為當前最優解。列舉完A班同學身高為最小值,再列舉B,C兩個班同學即可。

之前查詢是依次從最小值查詢直至找到第一個不小於當前身高,這樣時間複雜度為O(N²),超時。因此查詢使用的STL的lower_bound函式實現的,它內部是用二分查詢實現的,AC。

另外,由於A,B,C均為排了序的也可以用三個標記分別標記A,B,C三個陣列中當前遍歷的數,查詢B,C兩個班第一個不小於A的同學時不需要從頭遍歷,只需要從上次找到的值往後查詢即可,這樣時間複雜度降到了O(N)。

參考程式碼如下:

#include <iostream>
#include <algorithm>

using namespace std;

int a[100005]={0}, b[100005]={0}, c[100005]={0};
int n, m, l;

int minD(int a[], int b[], int c[], int n, int m, int l)  //a組成員的身高最低
{
	int ans=-1;
	for(int i=0; i<n; i++)
	{
		int bb=-1, cc=-1;
		//for(int j=0; j<m; j++)
		//{
		//	if(b[j]>=a[i])
		//	{
		//		bb = b[j];
		//		break;
		//	}
		int* pos;
		pos = lower_bound(b, b+m, a[i] );
		if(pos != b+m)
			bb = *pos;

		pos = lower_bound(c, c+l, a[i] );
		if(pos != c+l)
			cc = *pos;

		if(bb!=-1 && cc!=-1)
		{
			if(ans == -1)
				ans = 2 * max(bb-a[i], cc-a[i]);
			else
				ans = min(ans, 2 * max(bb-a[i], cc-a[i]));
		}
	}

	//for(int j=0; j<l; j++)
	//{
	//	if(c[j]>=a[i])
	//	{
	//		cc = c[j];
	//		break;
	//	}
	//}



	return ans;
}

int main()
{
	int result=-1, tt;
//	freopen("in.txt", "r", stdin);
	cin>>n>>m>>l;
	for(int i=0; i<n; i++)
		cin>>a[i];
	for(int i=0; i<m; i++)
		cin>>b[i];
	for(int i=0; i<l; i++)
		cin>>c[i];

	sort(a, a+n);
	sort(b, b+m);
	sort(c, c+l);

	tt = minD(a, b, c, n, m, l);
	if (tt != -1)
	{
		if(result == -1)
			result = tt;
		else
			result = min(result, tt);
	}

	tt = minD(b, a, c, m, n, l);
	if (tt != -1)
	{
		if(result == -1)
			result = tt;
		else
			result = min(result, tt);
	}

	tt = minD(c, a, b, l, n, m);
	if (tt != -1)
	{
		if(result == -1)
			result = tt;
		else
			result = min(result, tt);
	}

	cout<<result;

	return 0;
}