PAT甲级 1044 Shopping in Mars 二分法

使用upper_bound简化代码(从[start,end-1]内找出第一个大于该值的地址)

方法1

完全参照https://www.cnblogs.com/ligen/p/4351417.html的思路及代码

思路：

1.二分遍历找出满足条件的最小花费nears
2.二分遍历枚举下标i，输出所有花费恰好为最小花费nears的情况。

疑问

Q:判等条件为什么是sum[j-1]-sum[i]==s
A:因为Upper_bound返回第一个大于该值的，不包括等于该值。

#include <iostream>
#include<cstdio>
#include<algorithm>>
/* run this program using the console pauser or add your own getch, system("pause") or input loop */
using namespace std;

const int N = 100000+10;
int sum[maxn];
int n, s, nears = 1e8;

int main(int argc, char** argv) {
	scanf("%d %d", &n, &s);
	sum[0] = 0;
	for(int i = 1; i <= n; i++){
		scanf("%d", &sum[i]);
		sum[i] += sum[i-1];
	}
	//sum下标:1-n 
	//找到所有情况里花费的最小钱数 
	for(int i = 1; i <= n; i++){
		//jj为第一个和>=sum[i-1]+s的sum数组下标.即i-j的和满足的条件 
		int* jj = upper_bound(sum+i, sum+n+1, sum[i-1]+s);
		//从数组的begin位置到end-1位置二分查找 >= num的数字 
		//找到返回该数字地址，不存在则返回end 
		//通过返回地址-数组起始地址得到数字在数组中下标 
		int j = jj-sum;//j是数组下标
		if(sum[j-1]-sum[i-1] == s){
			nears = s;
			break;
		}else if(j <= n && sum[j] - sum[i-1] < nears){
			nears = sum[j] - sum[i-1];
		}
		//j成功找到没有越界(j<=n)且花费更小
		//大了， 
	}
	//然后再次二分查找出所有花费恰好为nears的情况并输出 
	for(int i = 1; i <= n; i++){
		int* jj = upper_bound(sum+i, sum+n+1, sum[i-1]+nears);
		int j = jj-sum;
		if(sum[j-1]-sum[i-1] == nears){
			printf("%d-%d\n", i, j-1);
		}
	}
	return 0;
}

方法2：

思路

用vector数组ans保存pair<int,int>记录i和j
由于upper_bound(sum+1, sum+1+n, sum[i-1]+m)只能返回第一个大于所给值的地址，导致就算能够恰好花费m元，它返回的也是花费m元的下标的后一个下标，这个思路需要分情况讨论，思维上需要缜密。
1.sum[j-1]-sum[i-1] == m ：若从下标i开始可以找到恰好为m
1）第一次找到恰好为m的
清空vector数组，再push
2)不是第一次
直接往vector数组里Push即可
2. j <= n&&sum[j]-sum[i-1] < nears :不能恰好为m,但是找到了目前为止更小的花费
3. j <= n && sum[j]-sum[i-1] == nears :不能恰好为m，该花费正好和之前最小花费相同

#include <iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<vector>
#include<string>
#include <map>
#define LL long long
using namespace std;
typedef pair<int, int> pr;

const int maxn = 100000+5;

int sum[maxn];
vector<pr> ans;


int main(){
    int n, m;
    scanf("%d %d", &n, &m);

    for(int i = 1; i <= n; i++){
        int x;
        scanf("%d", &sum[i]);
        sum[i] += sum[i-1];
    }
    int nears = 1e8;

    for(int i = 1; i <= n; i++){
        //查找范围为[begin, end-1]
        int* jj = upper_bound(sum+i, sum+n+1, sum[i-1]+m);
        int j = jj-sum;//数组下标+1
        //目前新的最小花费
        if(sum[j-1]-sum[i-1] == m){
            if(nears>m){
                nears = m;
                ans.clear();
                ans.push_back(make_pair(i,j-1));
            }else if(nears == m){
                ans.push_back(make_pair(i,j-1));
            }
        }else if(j <= n && sum[j]-sum[i-1] < nears){
            ans.clear();
            ans.push_back(make_pair(i, j));
            nears = sum[j] - sum[i-1];
        }else if(j <= n && sum[j] - sum[i-1] == nears){
            ans.push_back(make_pair(i, j));
        }
    }
    for(int i = 0; i < ans.size(); i++){
        printf("%d-%d\n", ans[i].first, ans[i].second);
    }
    return 0;
}