贪心法求解田忌赛马问题

问题描述

二千多年前的战国时期，齐威王与大将田忌赛马。双方约定每人各出300匹马，并且在上、中、下三个等级中各选一匹进行比赛，由于齐威王每个等级的马都比田忌的马略强，比赛的结果可想而知。现在双方各n匹马，依次派出一匹马进行比赛，每一轮获胜的一方将从输的一方得到200银币，平局则不用出钱，田忌已知所有马的速度值并可以安排出场顺序，问他如何安排比赛获得的银币最多。
输入：输入包含多个测试用例，每个测试用例的第一行正整数n（n≤1000）马的数量，后两行分别是n个整数，表示田忌和齐威王的马的速度值。输入n=0结束。
输出：每个测试用例输出一行，表示田忌获得的最多银币数。

问题求解

田忌的马速度用数组a表示，齐威王的马速度用数组b表示，将a、b数组递增排序。采用常识性的贪心思路，分以下几种情况：
（1）田忌最快的马比齐威王最快的马快即a[righta]>b[rightb]，则两者比赛（两个最快的马比赛），田忌赢。因为此时田忌最快的马一定赢，而选择与齐威王最快的马比赛对于田忌来说是最优的。
（2）田忌最快的马比齐威王最快的马慢即a[righta]<b[rightb]，则选择田忌最慢的马与齐威王最快的马比赛，田忌输。因为齐威王最快的马一定赢，而选择与田忌最慢的马比赛对于田忌来说是最优的。
（3）田忌最快的马与齐威王最快的马的速度相同即a[righta]=b[rightb]，又分为以下3种情况：
① 田忌最慢的马比齐威王最慢的马快即a[lefta]>b[leftb]，则两者比赛（两个最慢的马比赛），田忌赢。因为此时齐威王最慢的马一定输，而选择与田忌最慢的马比赛对于田忌来说是最优的。
② 田忌最慢的马比齐威王最慢的马慢，并且田忌最慢的马比齐威王最快的马慢，即a[lefta]≤b[leftb]且a[lefta]<b[rightb]，则选择田忌最慢的马与齐威王最快的马比赛，田忌输。因为此时田忌最慢的马一定输，而选择与齐威王最快的马比赛对于田忌来说是最优的。
③ 其他情况，即a[righta]=b[rightb]且a[lefta]≤b[leftb]且a[lefta]≥b[rightb]，则a[lefta]≥b[rightb]=a[righta]，即a[lefta]=a[righta]，b[leftb]≥a[lefta]=b[rightb]，即b[leftb]=b[rightb]，说明比赛区间的所有马的速度全部相同，任何两匹马比赛都没有输赢。

代码

int n;
int a[MAXN];
int b[MAXN];
int ans;

void solve()
{
	sort(a, a + n);
	sort(b, b + n);
	ans = 0;
	int lefta = 0, leftb = 0;
	int righta = n - 1, rightb = n - 1;

	while (lefta <= righta)
	{
		if (a[righta] > b[rightb])//田忌最快的马比齐威王最快的马快
		{
			ans += 200;
			righta--;
			rightb--;
		}
		else if (a[righta] < b[rightb])//田忌最快的马比齐威王最快的马慢
		{
			ans -= 200;
			lefta++;
			rightb--;
		}
		else
		{
			if (a[lefta] > b[leftb])//田忌最慢的马比齐威王最慢的马慢
			{
				ans += 200;
				lefta++;
				leftb++;
			}
			else
			{
				if (a[lefta] < b[leftb])
					ans -= 200;
				lefta++;
				rightb--;
			}
		}
	}
}

算法分析

算法的主要时间花费在排序上，时间复杂度为O(nlog2n)。