leetcode 刷题之路 7 Two Sum

Given an array of integers, find two numbers such that they add up to a specific target number.

The function twoSum should return indices of the two numbers such that they add up to the target, where index1 must be less than index2. Please note that your returned answers (both index1 and index2) are not zero-based.

You may assume that each input would have exactly one solution.

Input: numbers={2, 7, 11, 15}, target=9
Output: index1=1, index2=2

这道题可以分解为两个部分,一个为寻找升序数组中和为某个值的两个元素,这个问题已经在这篇文章中解答过了,另外一个就是记录这两个数字对应下标的问题。我采用pair类将每个数字和其对应的下标值合并存储。得到一个pair<int,int>类型的数组。具体的做法为,对这个数组按照数字大小进行稳定的排序,这一步直接采用STL中的stable_sort实现,在这个有序数组的基础上再使用上面文章链接里面给出的方法求出两个元素,从而也获得了对应的下标,问题解决。

my accepted answer:

bool cmp(pair<int, int> pii1, pair<int, int> pii2)
{
	return pii1.first < pii2.first;
}
class Solution {
public:
    vector<int> twoSum(vector<int> &numbers, int target) 
    {
        vector<pair<int, int>> vpii;
		for (int i = 0; i < numbers.size(); i++)
			vpii.push_back(pair<int,int>(numbers[i], i));
		
		stable_sort(vpii.begin(), vpii.end(), cmp);

		vector<int> vi;
		int i = 0, j = vpii.size() - 1;
		while (i < j)
		{
			if (vpii[i].first + vpii[j].first == target)
			{
		 		vi.push_back(vpii[i].second+1);
				vi.push_back(vpii[j].second + 1);
				sort(vi.begin(), vi.end());
				break;
			}
			else if (vpii[i].first + vpii[j].first>target)
				j--;
			else i++;
		}
		return vi;
    }
};


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值