题目
给定一串数字,和一个目标值,要求求出一组子序列和等于目标值的一对儿下标吗,如果没有相等的,找到一个大于目标值的最小值。
思路一 暴力 会超时
- 直接遍历每个个可能性
- 中间可以提前跳出的,就是条件多一些,,有很多if。。整体代码写的不好。
- 有两个点会超时,有一个点答案错误,应该也和这个方法有关
#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; }
思路二 二分法
- 二分法查找,因为这串数字都是正值,所以设定数组sum[i],表示从1 ~ i的和
- sum数组是一定递增的,所以可以使用二分查找
- 枚举左端点i,查找在i ~ n 值为sum[i - 1] + S 的值(值的来源为sum[j] - sum[i - 1] = s)是否存在,如果存在,则把对应的下标作为右端点j,如果不存在,找到第一个使得和值超过s的右端点j。
- 需要对序列进行两次遍历,第一次遍历找到题目所需要的和(等于的话就是目标值K,没有等于的话就是大于k的最小值neark)
- 第二次遍历就是在确定值的情况下,寻找下标,这次遍历是一定得到确定的值的
#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; }