编程之美 2.19 区间重合判断

题目大意：

给定源区间[x, y] (x≤y) 和 n个无序的目标区间[xi, yi](1 ≤ i ≤ n)。

判断源区间是否在目标区间内。

分析：

书中介绍了两种解法。

第一种：将目标区间投影到源区间，看最终未被覆盖的区间是否变为{∅}。

这种方法的复杂度无疑是很高的。 O(n^2)。

另外，对于K组源区间查询，其复杂度是单次的K倍。

第二种："线段树"

这种解法很好解决了第一种算法出现的问题。

先将所有n组目标区间按照X排序，并合并为多个小区间。【这些小区间是有序的】

之后对于给定的源区间，基于二分思想定位区间的位置。

复杂度：

目标区间排序：O(n*logn)

排序后目标区间合并：O(n)

源区间一次查询：O(logn)

因此

对于K组源区间查询

总体复杂度：O(nlogn + Klogn)。

大大提高了效率。

下面给出自己实现的代码：

【已在VS2013上编译测试通过】

#include <iostream>
#include <vector>
#include <algorithm>

using namespace std;

typedef pair<int, int> Range;

bool cmp(const Range& a, const Range& b)
{
	return a.first < b.first;
}

bool binarySearch(vector<Range> &mergeRange, Range &src)
{
	bool isFound = false;
	int l = 0, r = mergeRange.size() - 1;
	int mid;

	while (l <= r)
	{
		mid = l + (r - l) / 2;

		// 源区间正好在某个子区间内
		if (mergeRange[mid].first <= src.first && mergeRange[mid].second >= src.second)
		{
			isFound = true;
			break;
		}
		else if (mergeRange[mid].second <= src.first)
		{
			l = mid + 1;
		}
		else if (mergeRange[mid].first >= src.second)
		{
			r = mid - 1;
		}
		else
		{
			isFound = false;
			break;
		}
	}

	return isFound;
}

int main(void)
{
	Range src, dest;

	vector<Range> vRs;

	cout << "请输入目标区间 -1 -1 为结束标志" << endl;
	while (cin >> dest.first >> dest.second, dest.first != -1 && dest.second != -1)
	{
		vRs.push_back(dest);
	}

	// 1. 排序目标区间
	sort(vRs.begin(), vRs.end(), cmp);

	// 2. 合并目标区间
	vector<Range> mergeRange;
	vector<Range>::iterator it = vRs.begin();
	int st = it->first, ed = it->second;

	for (; it != vRs.end(); ++it)
	{
		// 前n-1个区间
		if ((it + 1) != vRs.end())
		{
			if ((it + 1)->first > ed)
			{
				mergeRange.push_back(make_pair(st, ed));
				st = (it + 1)->first;
			}
			ed = ((it + 1)->second > ed) ? (it + 1)->second : ed;
		}
		// 剩下最后一个区间
		else
		{
			if (it->first > ed)
			{
				mergeRange.push_back(make_pair(st, ed));
				mergeRange.push_back(*it);
			}
			else
			{
				ed = ((it)->second > ed) ? (it)->second : ed;
				mergeRange.push_back(make_pair(st, ed));
			}
		}
	}

	cout << "请输入要查找的源区间 -1 -1 为结束标志" << endl;
	while (cin >> src.first >> src.second, src.first != -1 && src.second != -1)
	{
		// 3. 查找是否在目标区间内
		if (binarySearch(mergeRange, src))
		{
			cout << src.first << " " << src.second << " is included in dest ranges" << endl;
		}
		else
		{
			cout << src.first << " " << src.second << " is NOT included in dest ranges" << endl;
		}
	}

	return 0;
}

下面给出运行截图：

对于文中给出的扩展问题

改天再更新~

希望大家批评指正~