码迷,mamicode.com
首页 > 其他好文 > 详细

PAT A1044 Shopping in Mars [二分]

时间:2019-08-06 21:32:17      阅读:130      评论:0      收藏:0      [点我收藏+]

标签:line   i++   main   就是   turn   end   链接   保存   tps   

题目描述

链接
求一串的数字中连续的一段,使得这个连续的段内数字的和恰好等于所期望的值m。如果不能找到恰好等于,就找让自己付出最少的价格(总和必须大于等于所给值)的那段区间。求所有可能的结果

分析

  • 输出区间和等于指定值的方案,可以先统计前缀和,然后作差就可以得到区间和
  • 原本错误的做法:作差得到区间和,保存在node数组里面,然后再排序,然后再二分查找。实际没用,因为最大的复杂度是\(O(n^2)\)
  • 正解应该是:因为本来前缀和就是递增的,可以枚举区间左端点,再二分查找区间右端点,使得区间和=m,复杂度\(O(nlogn)\)
//从错误做法里面还学到了内存超限的解决方法,用resize精确控制
#include<bits/stdc++.h>
using namespace std;


int n,m;
int a, tmp;
struct node{
    int l,r;
    int sum;
};
vector<node> rec;
vector<int> b;
bool cmp(node a, node b){
    if(a.sum != b.sum) return a.sum < b.sum;
    return a.l < b.l;
}

int main(){
    cin>>n>>m;
    b.resize(n+1); //解决内存超限
    for(int i=1;i<=n;i++){
        cin>>a;
        tmp += a;
        b[i] = tmp;
    }
    for(int i=0;i<=n;i++){
        for(int j=i;j<=n;j++){
            node tmp = {i+1,j,b[j]-b[i]};
            rec.push_back(tmp);
        }
    }
    sort(rec.begin(),rec.end(),cmp);
    int l = 0,r = rec.size();
    while(l<r){
        int mid = (l+r)>>1;
        if(rec[mid].sum >= m) r = mid;
        else l = mid + 1;
    }
    if(rec[l].sum > m){
        int delta = rec[l].sum - m;
        while(rec[l].sum-m == delta){
            cout<<rec[l].l<<"-"<<rec[l].r<<endl;
            l++;
        }
    }else{
        while(rec[l].sum == m){
            cout<<rec[l].l<<"-"<<rec[l].r<<endl;
            l++;
        }
    }

}

PAT A1044 Shopping in Mars [二分]

标签:line   i++   main   就是   turn   end   链接   保存   tps   

原文地址:https://www.cnblogs.com/doragd/p/11311868.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!