问题:
1. 给定一个源区间[x,y]和N个无序的目标区间[x1,y1] [x2,y2] … [xn,yn],判断源区间[x,y]是不是在目标区间内。
2. 给定一个窗口区域和系统界面上的N个窗口,判断这个窗口区域是否被已有的窗口覆盖。
1. 解法:
先用区间的左边界值对目标区间进行排序O(nlogn),对排好序的区间进行合并O(n),对每次待查找的源区间,用二分查出其左右两边界点分别处于合并后的哪个源区间中O(logn),若属于同一个源区间则说明其在目标区间中,否则就说明不在。
#include <iostream> #include <algorithm> using namespace std; struct Line { int low, high; bool operator<(const Line &l) const {return low<l.low;} }; #define MAXN 10001 Line lines[MAXN]; // 目标区间 int ncnt = 0; // 合并后区间的个数 #define N 101 Line sl[N]; // 待查询的源区间 // 用二分查找找出key所在的区间,以区间的low作为划分 int GetIndex(int key) { int u, v; u = 0; v = ncnt-1; while (u<=v) // u,v可取等号 { int m = (u+v)>>1; if (key >= lines[m].low) u = m+1; else v = m-1; } return v; } int main() { int n, k, i, j; cin >> n >> k; // n是目标区间的个数,k是待查询的源区间的个数 for (i=0; i<n; i++) cin >> lines[i].low >> lines[i].high; for (i=0; i<k; i++) cin >> sl[i].low >> sl[i].high; // 排序O(nlogn) sort(lines, lines+n); // 合并O(n) int lasthigh = lines[0].high; for (i=1; i<n; i++) if (lasthigh >= lines[i].low) lasthigh = lines[i].high; else { lines[ncnt++].high = lasthigh; lines[ncnt].low = lines[i].low; lasthigh = lines[i].high; } lines[ncnt++].high = lasthigh; for (i=0; i<k; i++) { // 单词查找时间O(logn) int s1 = GetIndex(sl[i].low); int s2 = GetIndex(sl[i].high); if (s1==s2 && sl[i].high <= lines[s2].high) printf("Yes\n"); else printf("No\n"); } }
2. 解法:
这个问题适合使用线段树来解答,单次查找的时间复杂度为O(nlogn),当然也能用数组解答,但单次查找的时间复杂度会增加到O(n^2)。这里我们直接使用线段树来解答。
线段树是一棵二叉树,将数轴划分成一系列的初等区间[I, I+1] (I=1,2,..,N-1)。每个初等区间对应于线段树的一个叶结点。线段树的内部结点对应于形如[ I, J ](J – I > 1)的一般区间。由于线段树给每一个区间都分配了结点,利用线段树可以求区间并后的总长度与区间并后的线段数。先给出测试数据(前4行是系统界面上已有的N个窗口,之后的一行是待测试的窗口区域),后面是代码:
4
-15 0 5 10
-5 8 20 25
15 -4 24 14
0 -6 16 4
2 15 10 22
#include <iostream>
#include <cmath>
#include <algorithm>
using namespace std;
// 线段树的结点
struct SegNode
{
int low, high; // 线段的两端点索引
int ncover; // 线段被覆盖的次数
SegNode *left; // 结点的左子树
SegNode *right; // 结点的右子树
SegNode() {low=high=0;ncover=0;
left=right=NULL;}
};
// 构造线段树,它是一个完全二叉树
void BuildSegTree(SegNode *&tree, int *index, int low, int high)
{
if (low < high)
{
tree = new SegNode;
tree->low = low;
tree->high = high;
if (high-low>1)
{
int m = (low+high)/2;
BuildSegTree(tree->left, index, low, m);
BuildSegTree(tree->right, index, m, high);
}
}
}
// 往线段树中插入线段,即用线段(low,high)来覆盖线段树
void InsertSegTree(SegNode *tree, int low, int high)
{
// 先序遍历
if (low<=tree->low && tree->high<=high)
tree->ncover++;
else if (tree->high-tree->low > 1)
{
int m = (tree->low+tree->high)/2;
if (low < m) InsertSegTree(tree->left, low, high);
if (m < high) InsertSegTree(tree->right, low, high);
}
}
// 从线段树中删除线段
void DeleteSegTree(SegNode *tree, int low, int high)
{
if (low<=tree->low && tree->high<=high)
tree->ncover--;
else if (tree->high-tree->low > 1)
{
int m = (tree->low+tree->high)/2;
if (low < m) DeleteSegTree(tree->left, low, high);
if (m < high) DeleteSegTree(tree->right, low, high);
}
}
// 线段树中是否包含线段(low,high)
bool FindSegTree(SegNode *tree, int low, int high)
{
// 若当前区间被覆盖,且线段(low,high)属于当前区间则返回覆盖
if (tree->ncover && tree->low <= low && high <= tree->high )
return true;
// 若(low,high)没被当前区间覆盖,则将其分为两段,
// 分别考虑是否被子结点表示的区间覆盖
else if (tree->high - tree->low > 1)
{
int m = (tree->low + tree->high) >> 1;
bool ret = true;
if (low<m) ret = FindSegTree(tree->left, low, high<m?high:m);
if (!ret) return false;
if (m<high) ret = FindSegTree(tree->right, m<low?low:m, high);
if (!ret) return false;
return true;
}
return false;
}
#define LEFT true
#define RIGHT false
#define INF 10000
// 表示竖直方向的线段
struct Line
{
int starty, endy; // 竖线的长度
int x; // 竖线的位置
bool inout; // 竖线是长方形的左边还是右边
bool operator<(const Line& a) const{ // 依据x坐标进行排序
return x<a.x;
}
};
// 所有竖直方向的线段
Line lines[INF];
// 对横向超元线段进行分组
int index[INF];
int nCnt = 0;
// 获取key的位置
int GetIndex(int key)
{
// 用二分查找查出key在index中的位置
return lower_bound(index,index+nCnt,key)-index;
}
// 获取key的位置或比它小的最大数的位置
int GetLower(int key)
{
size_t pos = lower_bound(index,index+nCnt,key)-index;
if (key == index[pos]) return pos;
else return pos-1;
}
// 获取key的位置或比它大的最小数的位置
int GetUpper(int key)
{
return lower_bound(index,index+nCnt,key)-index;
}
int main()
{
int nRec;
cin >> nRec;
int i, j;
int x[2], y[2];
// 读取nRec个窗口的数据
for (i=0; i<nRec; i++)
{
cin >> x[0] >> y[0] >> x[1] >> y[1];
// 记录每个长方形的两条竖直边
lines[2*i].x=x[0]; lines[2*i+1].x=x[1];
lines[2*i].starty=lines[2*i+1].starty=min(y[0],y[1]);
lines[2*i].endy=lines[2*i+1].endy=max(y[0],y[1]);
lines[2*i].inout=LEFT; lines[2*i+1].inout=RIGHT;
// 对竖直的线段进行离散化
index[2*i]=y[0]; index[2*i+1]=y[1];
}
// 待查询的窗口区域
Line search[2];
cin >> x[0] >> y[0] >> x[1] >> y[1];
search[0].x=x[0]; search[1].x=x[1];
search[0].starty=search[1].starty=min(y[0],y[1]);
search[0].endy=search[1].endy=max(y[0],y[1]);
search[0].inout=LEFT; search[1].inout=RIGHT;
// 对x坐标进行排序O(nlogn)
sort(index, index+2*nRec);
sort(lines, lines+2*nRec);
// 排除index数组中的重复数据O(n)
for (i=1; i<2*nRec; i++)
if (index[i]!=index[i-1])
index[nCnt++] = index[i-1];
index[nCnt++] = index[2*nRec-1];
// 建立线段树
SegNode *tree;
BuildSegTree(tree, index, 0, nCnt-1);
// 单词查找的时间复杂度为O(nlogn)
bool res;
InsertSegTree(tree, GetIndex(lines[0].starty), GetIndex(lines[0].endy));
for (i=1; i<2*nRec; i++)
{
if (lines[i].inout==LEFT) // 遇窗口的左边界,将其加入线段树
InsertSegTree(tree, GetIndex(lines[i].starty), GetIndex(lines[i].endy));
else // 遇窗口的右边界,将其删出线段树
DeleteSegTree(tree, GetIndex(lines[i].starty), GetIndex(lines[i].endy));
if (lines[i].x!=lines[i-1].x && search[0].x < lines[i+1].x && search[1].x > lines[i].x)
{
// 从待查窗口区域的左边界开始查询直到其右边界结束查询
res = FindSegTree(tree, GetLower(search[0].starty), GetUpper(search[0].endy));
if (!res) break;
}else if (search[1].x <= lines[i].x)
break;
}
if (res) printf("Yes\n");
else printf("No\n");
return 0;
}