[剑指Offer] 42.和为S的两个数字

题目描述

输入一个递增排序的数组和一个数字S，在数组中查找两个数，是的他们的和正好是S，如果有多对数字的和等于S，输出两个数的乘积最小的。 

输出描述:

对应每个测试案例，输出两个数，小的先输出。

【思路】同LeetCode中的Two Sum类似，但由于要将乘积最小的一组输出，因此需要一个辅助函数Sift来将不符合条件的删除。

class Solution
{
public:
    void Sift(vector<int> &vec)
    {
        int minid = 0;
        int num = vec.size();
        vector<int> multi(num/2);
        for(int i = 0,j = 0; i < num; i = i + 2,j ++)
        {
            multi[j] = vec[i]*vec[i+1];
        }
        for(int i = 1; i < num/2; i ++)
        {
            if(multi[minid] > multi[i])
            {
                vec[minid*2] = vec[minid*2+1] = -1;
                minid = i;
            }
            else
            {
                vec[i*2] = vec[i*2+1] = -1;
            }
        }
        for (vector<int>::iterator iter = vec.begin(); iter != vec.end(); )
        {
            if (*iter == -1)
            {
                if (++iter == vec.end())
                {
                    vec.erase(--iter);
                    break;
                }
                else iter = vec.erase(--iter);
            }
            else
                ++iter;
        }
    }

    vector<int> FindNumbersWithSum(vector<int> array,int sum)
    {
        vector<int> res;
        if(array.size() < 2) return res;
        int target = 0;
        vector<int> temp;
        for(int i = 0; i < array.size(); i ++)
        {
            target = sum - array[i];
            if(find(temp.begin(),temp.end(),target) != temp.end())
            {
                res.push_back(target);
                res.push_back(array[i]);
            }
            else
            {
                temp.push_back(array[i]);
            }
        }
        Sift(res);
        return res;
    }
};