PAT 1044

题目

给定一串数字，和一个目标值，要求求出一组子序列和等于目标值的一对儿下标吗，如果没有相等的，找到一个大于目标值的最小值。

思路一 暴力 会超时

1. 直接遍历每个个可能性
2. 中间可以提前跳出的，就是条件多一些，，有很多if。。整体代码写的不好。
3. 有两个点会超时，有一个点答案错误，应该也和这个方法有关

   #include<cstdio>
   #include<vector>
   using namespace std;
   struct cas {
   	int start;
   	int end;
   	cas(int _start, int _end) : start(_start), end(_end) {}
   };
   const int maxn = 100010;
   
   int k, n;
   int a[maxn];
   vector<cas> v;
   
   int main() {
   	scanf("%d %d", &n, &k);
   	for (int i = 1; i <= n; i++) {
   		scanf("%d", &a[i]);
   	}
   	int flag = 0;
   	int min = 100000;
   	for (int i = 1; i <= n; i++) {
   		int sum = a[i];
   		for (int j = i + 1; j <= n + 1; j++) {
   			if (sum == k) {
   				if (flag == 0) {
   					v.clear();
   					v.push_back(cas(i, i));
   					flag = 1;
   					break;
   				}
   				else {
   					v.push_back(cas(i, i));
   					break;
   				}
   			}
               if(j > n) break;
   			sum += a[j];
   			if (sum > k) {
   				if (flag == 0 && sum < min) {
   					min = sum;
   					v.clear();
   					v.push_back(cas(i, j));
   					break;
   				}
   				else if (flag == 0 && sum == min) {
   					v.push_back(cas(i, j));
   					break;
   				}
   				else break;
   			}
   			else if (sum == k) {
   				if (flag == 0) {
   					v.clear();
   					v.push_back(cas(i, j));
   					flag = 1;
   					break;
   				}
   				else {
   					v.push_back(cas(i, j));
   					break;
   				}
   			}
   		}
   	}
   	for (int i = 0; i < v.size(); i++) {
   		printf("%d-%d\n", v[i].start, v[i].end);
   	}
   	return 0;
   }
   

思路二 二分法

1. 二分法查找，因为这串数字都是正值，所以设定数组sum[i]，表示从1 ~ i的和
2. sum数组是一定递增的，所以可以使用二分查找
3. 枚举左端点i，查找在i ~ n 值为sum[i - 1] + S 的值（值的来源为sum[j] - sum[i - 1] = s）是否存在，如果存在，则把对应的下标作为右端点j，如果不存在，找到第一个使得和值超过s的右端点j。
4. 需要对序列进行两次遍历，第一次遍历找到题目所需要的和（等于的话就是目标值K，没有等于的话就是大于k的最小值neark）
5. 第二次遍历就是在确定值的情况下，寻找下标，这次遍历是一定得到确定的值的

   #include<cstdio>
   using namespace std;
   const int maxn = 100010;
   int sum[maxn];
   
   int n, k;
   int neark = 100000010;
   
   int upper_bound(int L, int R, int x) {
   	int left = L, right = R , mid;
   	while (left < right) {
   		mid = (left + right) / 2;
   		if (sum[mid] > x) {
   			right = mid;
   		}
   		else {
   			left = mid + 1;
   		}
   	}
   	return left;
   }
   
   int main() {
   	int n, k;
   	scanf("%d %d", &n, &k);
   	sum[0] = 0;
   	for (int i = 1; i <= n; i++) {
   		scanf("%d", &sum[i]);
   		sum[i] += sum[i - 1];
   	}
   	for (int i = 1; i <= n; i++) {
   		int j = upper_bound(i, n + 1, sum[i - 1] + k);
   		if (sum[j - 1] - sum[i - 1] == k) {
   			neark = k;
   			break;
   		}
   		else if (j <= n && sum[j] - sum[i - 1] < neark) {
   			neark = sum[j] - sum[i - 1];
   		}
   	}
   	for (int i = 1; i <= n; i++) {
   		int j = upper_bound(i, n + 1, sum[i - 1] + neark);
   		if (sum[j - 1] - sum[i - 1] == neark) {
   			printf("%d-%d\n", i, j - 1);
   		}
   	}
   	return 0;
   }