练习思路整理

1.第 k 小子序列和
先把所有数从小到大排序.
设一个二元组 (sum,i) , 表示以第 i 位为结尾, 和为 sum 的子序列, 建一个小根堆来维护这些二元组, 每次取出堆顶, 然后把 (sum+ai+1,i+1) 和 (sum+ai+1−ai,i+1) 加入堆中, 即把第 i+1 个数加入到序列中, 或把序列中最大的数替换成第 i+1 个数. 显然这样可以不重不漏地按顺序遍历完所有情况.

然后第 k 个被取出堆顶的数就是答案.

2.是否是素数

bool IsPrime(int n)
{
	if (n == 2 || n == 3) return true;
	if (n % 6 != 1 && n % 6 != 5)
		return false;
	for (int i = 5; i <= sqrt(n); i += 6)
	{
		if (n % i == 0 || n % (i + 2) == 0)
			return false;
	}
	return true;
}