题目大意:给出一个正数序列,在当中找到满足这样条件的连续子串:子串的和大于等于给定值。输出具有最小的和的子串的开始和结束位置的下标(从1开始计),如果这样的子串有两个以上,按照开始位置从小到大,输出所有子串的开始和结束位置。
一开始直接暴力遍历所有子串,并维护最小的子串和。但这样还是有两个测试点超时,开始位置肯定是需要遍历序列的每个元素的,那么减少耗时的主要方法就是如何查找结束位置,问题就转换为:固定开始位置,找到结束位置,使得这个子串的和是从开始位置到整个序列结尾的最小的大于等于M的数。应该是用二分来查找,这样就首先要在输入时,记录每个位置相对于位置1的子串和,那么从位置 i 到 位置 j 的和可以表示为 sumFromFirst[ j ] - sumFromFirst [ i - 1 ]. 之后就是遍历 i, 用二分查找 j。但这里还有一个问题,找到的 j 并不一定是 和等于 M的位置,如果和不能等于M,需要输出大于M的最小值的位置。所以这里用一个数组,来存储二分过程中所有和大于等于M的位置,那么要求的应当是最后一个加入数组的元素,无论最后的和是等于M还是大于M。遍历 i 的同时,维护最小的子串和。最后输出。
AC代码:
#include <vector>
#include <cstdio>
#include <map>
using namespace std;
void search(vector<int> &sumFromFirst, int left, int right, int M, int i, vector<int> &upperBound)
{
if(left > rig