剑指offer 41、42:和为S的连续正数序列 、和为S的两个数字

41:题目描述
小明很喜欢数学,有一天他在做数学作业时,要求计算出9~16的和,他马上就写出了正确答案是100。但是他并不满足于此,他在想究竟有多少种连续的正数序列的和为100(至少包括两个数)。没多久,他就得到另一组连续正数和为100的序列:18,19,20,21,22。现在把问题交给你,你能不能也很快的找出所有和为S的连续正数序列? Good Luck!
输出描述:
输出所有和为S的连续正数序列。序列内按照从小至大的顺序,序列间按照开始数字从小到大的顺序
从1开始累加,如果大于sum则sum-start;++start,即从小值减

class Solution {
public:
    vector<vector<int> > FindContinuousSequence(int sum) {
        vector<vector<int> >data;
        if(sum<3)
            return data;
        int start=1,end=1;
        int s=1;
        while(start<=end)
        {
            ++end;
            s+=end;
            while(s>sum)
            {
                s-=start;
                ++start;
            }
            if(s==sum && start<end)
            {
                vector<int>temp_data;
                for(int i=start;i<=end;++i)
                    temp_data.push_back(i);
                data.push_back(temp_data);
            }

        }
        return data;
    }
};
# -*- coding:utf-8 -*-
class Solution:
    def FindContinuousSequence(self, tsum):
        # write code here
        data=[]
        if tsum<3:
            return data
        start=end=1
        s=1
        while start<=end:
            end+=1
            s+=end
            while s>tsum:
                s-=start
                start+=1
            if s==tsum and start != end:
                data.append(range(start,end+1))
        return data

42.题目描述
输入一个递增排序的数组和一个数字S,在数组中查找两个数,使得他们的和正好是S,如果有多对数字的和等于S,输出两个数的乘积最小的。
输出描述:
对应每个测试案例,输出两个数,小的先输出。
思路:两个数差越大,乘越小,故从两端开始遍历

class Solution {
public:
    vector<int> FindNumbersWithSum(vector<int> array,int sum) {
        vector<int>data;
        if(array.empty())
            return data;
        for(int i=0,j=array.size()-1;i<j;)
        {
            int temp_sum=array[i]+array[j];
            if(temp_sum>sum)
                --j;
            else if (temp_sum<sum)
                ++i;
            else
            {
                data.push_back(array[i]);
                data.push_back(array[j]);
                break;
            }
        }
        return data;
    }
};

# -*- coding:utf-8 -*-
class Solution:
    def FindNumbersWithSum(self, array, tsum):
        data=[]
        i=0
        j=len(array)-1
        while i<j:
            temp=array[i]+array[j]
            if temp>tsum:
                j-=1
            elif temp<tsum:
                i+=1
            else:
                data.extend([array[i],array[j]])
                break
        return data
        # write code here

猜你喜欢

转载自blog.csdn.net/zd_nupt/article/details/81448475