PAT（甲级）1044 Shopping in Mars (25point(s))

题目

题目链接

思路

题目大意：找出数组片段和大于等于目标值的片段；
因为题目要求如果没有片段和等于目标值的话输出大于目标值的最小片段，所以第一步要计算出与目标值相差最小的near是多少；
枚举i 从1到n，用二分找到大于等于 sum[i - 1] + m 的下标 j，如果某一个i满足 sum[i - 1] + m = sum[j]，则说明可以找到差值为m的位置，直接中断；
找到这个最小差值后，再一次二分查找一次，找到 大于等于 sum[i - 1] + near 的第一个位置，如果正好等于的话，就输出 i 和 j;
注意这是二分搜索的变种：找到大于等于目标值的第一个位置，迭代结束的条件是 i == j,如果找不到目标值的话，返回 n + 1;所以传入传入的参数是 n + 1;

代码

#include <iostream>
#include <string>
#include <cstring>
#include <vector>
#include <algorithm>
using namespace std;
const int maxn = 1e5 + 10;
int nums[maxn], sum[maxn];

//注意这是二分搜索的变形，找到第一个满足要求数的下标
int binarySearch(int left, int right, int x){
     int mid;
     while(left < right){
          mid = (left + right) / 2;
          if(sum[mid] >= x) right = mid;
          else left = mid + 1;
     }
     return left;//注意如果找不到那么返回最右边界加一的位置
}

int main()
{
     int n, m, nearSum = maxn;
     scanf("%d%d", &n, &m);
     sum[0] = 0;
     //计算和
     for(int i = 1; i <= n; i ++){
          scanf("%d", &nums[i]);
          sum[i] = sum[i - 1] + nums[i];
     }
     //找到与目标值最小的差值
     for(int i = 1; i <= n; i ++){
          //找到第一个大于等于目标值的位置
          int j = binarySearch(i, n + 1, sum[i - 1] + m);
          //如果找不到大于等于的位置会返回n + 1，这里我们并没有处理，因为不会对nearSun产生影响
          if(j <= n && sum[j] - sum[i - 1] == m) {
               nearSum = m;
               break;//说明最小值是m,已经不需要找了
          }
          else if(j <= n) nearSum = min(nearSum, sum[j] - sum[i - 1]);
     }
     //把能找到最小差值的位置输出
     for(int i = 1; i <= n; i ++){
         //注意右边界是n + 1
          int j = binarySearch(i, n + 1, sum[i - 1] + nearSum);
          //这里的j可能大于等于，我们只输出等于的情况
          if(j <= n && sum[j] == nearSum + sum[i - 1]){
               printf("%d-%d\n", i, j);
          }
     }
     system("pause");
     return 0;
}