贪心算法之 POJ 2313 Sequence

对于此题首先对V的定义换种形式来写

V = (|A(1) – B(1)| + |A(2) – B(2)| + ... + |A(N) – B(N)|) + (|B(1) – B(2)| + |B(2) – B(3)| + ... +|B(N-1) – B(N)|)

对于规模为K时可以写成；

A1 A2 A3 A4 。。。 AK-1 AK

V =  |A1  - B1 | + | A2 - B2 | +| A3 - B3 | +| A4 - B4| 。。。+| AK-1 - BK-1| +|AK - BK |

+ |B2 - B1 | +|B3 - B2 | +| B4 - B3 | 。。。+|BK-1 - BK-2 | +|BK-BK-1|

如果把集合A和B看作数轴上的点，V就是线段A2B1 A3B2 A4B3 。。。 AK-1BK-2 AKBK-1线段的长度

观察可以发现对于Bi ，对于线段AiBi-1的长度是无影响的，只要Bi在线段AiBi-1范围内取最接近点Ai+1的值，则线段Ai+1Bi取得最小长度，问题规模扩展到了K+1。

到此我们差不多找出了贪心算法递推关系式：

Vk+1 = Vk + |Ak+1 - Bk|

Bk =   x  ；x满足 min { |    | Ak+1 - x | }   x在范围  Ak到Bk-1 之间（假设Ak > Bk-1，反之同理）

B1 = A1；

下面对递推关系进行证明：

显然 n = 1 时 V1 = 0 成立；

下面传照片了，编辑电子版太慢了

代码：

#include <iostream>

using namespace std;

/*
*根据Bi,Ai+1,Ai+2,计算Bi+1的值
*/

int find(int bi, int aii, int aiii )
{
	int max,min;
	//找出Bi,Ai+1,定义的线段范围
	if( bi > aii )
	{
		min = aii;
		max = bi;
	}
	else
	{
		min = bi;
		max = aii;
	}

	//如果Ai+2在线段左边
	if( aiii < min )
		return min;
	//如果Ai+2在线段右边
	if( aiii > max )
		return max;

	//如果Ai+2在线段内部
	return aiii;
}

int main()
{
	int num;
	cin>>num;
	//bi存储Bi，aii存储Ai+1，aiii存储Ai+2，sum存储V的值
	int bi,aii,aiii,sum=0;
	//如果输入为一个数，B1=A1，V=0
	if( num == 1 )
	{
		cin>>bi;
		cout<<0<<endl;
		return 0;
	}
	//如果输入2个数，V = 线段A2B1，,B1 = A1 
	if( num == 2 )
	{
		cin>>bi;
		cin>>aii;
		sum = bi > aii ? ( bi - aii ) : ( aii - bi );
		cout<<sum<<endl;
		return 0;
	}
	//如果输入大于等于三个数
	//初始B1=A1
	cin>>bi>>aii;
	//V  +=  线段 A2B1
	sum += bi > aii ? ( bi - aii ) : ( aii - bi );
	//加上余下num-2条线段的长度
	num -= 2;
	while( num-- > 0 )
	{
		//aiii存储Ai+2
		cin >> aiii;
		//根据Ai+2更新bi为Bi+1的值作为本次循环中的Bi
		bi = find(bi,aii,aiii);
		//aii更新为Ai+2的值作为下次循环的Ai+1
		aii = aiii;
		//sum += 线段Ai+1Bi的值
		sum += bi > aii ? ( bi - aii ) : ( aii - bi );
	}
	cout<<sum<<endl;
	return 0;
}