3Sum

第一次写用了dfs,做了一定优化,但是还是TLE

原来这种题有套路解法,用两个游标相对寻找会节省很多时间

class Solution {
public:
	vector> threeSum(vector& nums) {
		set> res;//set不允许存储相同的元素,达到去重的效果
		sort(nums.begin(), nums.end());//压缩时间复杂度第一处

		for (int k = 0; k < nums.size(); ++k) {
			if (nums[k] > 0) break;//处理corner case,同时也是压缩时间复杂度的第二处

			int target = 0 - nums[k];
			int i = k + 1, j = nums.size() - 1;

			while (i < j) {
				if (nums[i] + nums[j] == target) {
					res.insert({ nums[k], nums[i], nums[j] });

					while (i < j && nums[i] == nums[i + 1]) ++i;//去重,压缩时间复杂度第三处
					while (i < j && nums[j] == nums[j - 1]) --j;
					++i; --j;
				}
				else if (nums[i] + nums[j] < target) ++i;
				else --j;
			}
		}
		return vector>(res.begin(), res.end());
	}
};


你可能感兴趣的:(leetcode)