小米笔试题 线段覆盖的长度

昨天去小米笔试了，其中一道大题是求一个字符串的所有排列，如“AAB”的所有排列为{“AAB”、“ABA”，“BAA”}，这道题在我之前的博客中有研究过，所以不赘述，请查看这里。

另外一道题是给定一些线段，线段有起点和终点，求这些线段覆盖的距离，重复的部分只计算一次。这道题其实应该也算比较easy了，昨天笔试的时候，写得有点复杂了。今天整理了一下思路：首先排序，起点低的在前面，起点相同的按终点排。然后在进行扫描，并求距离。不是很好描述，下面是代码：

#include 
#include 

using namespace std;

struct Segment
{
	int start;
	int end;
};

bool cmp(const Segment &seg1, const Segment &seg2)
{
	if (seg1.start < seg2.start)
	{
		return true;
	}
	else if (seg1.start > seg2.start)
	{
		return false;
	}
	else
	{
		return seg1.end < seg2.end;
	}
}


int sum(int n, Segment *segs)
{
	sort(segs, segs + n, cmp);
	int line = 0, start, end;
	for (int i = 0; i < n; i++)
	{
		if (i == 0 || segs[i].start > end)
		{
			start = segs[i].start;
			end = segs[i].end;
			line += end - start;
		}
		else if (segs[i].end > end)
		{
			line += segs[i].end - end;
			end = segs[i].end;
		}
	}
	return line;
}

int main()
{
	Segment segs[3];
	segs[1].start = 1;
	segs[1].end = 2;
	segs[0].start = 1;
	segs[0].end = 3;
	segs[2].start = 5;
	segs[2].end = 6;
	int s = sum(3, segs);
	cout << s << endl;
	system("pause");
	return 0;
}