这道题目,我的解法又是过于复杂了,用了map,而且把情形进行分类,结果搞得非常复杂。我的代码如下:
vector<int> getVec(int a, int b, int c) {
vector<int> result;
int arr[] = {a, b, c}, tmp;
for (int i=0; i<2; i++) {
for (int j=i+1; j<3; j++) {
if (arr[i] > arr[j]) {
tmp = arr[i];
arr[i] = arr[j];
arr[j] = tmp;
}
}
}
result.push_back(arr[0]);
result.push_back(arr[1]);
result.push_back(arr[2]);
return result;
}
vector<vector<int>> threeSum(vector<int>& nums) {
map<int, int> num_cnt;
vector<vector<int>> result;
vector<int> item;
int i, j, sum;
// 把数字放进map
for (i=0; i<nums.size(); i++) {
if (num_cnt.find(nums.at(i)) != num_cnt.end()) {
num_cnt[nums.at(i)] = num_cnt[nums.at(i)] + 1;
} else {
num_cnt[nums.at(i)] = 1;
}
}
nums.clear();
for (map<int, int>::iterator it = num_cnt.begin(); it != num_cnt.end(); it++) {
nums.push_back(it->first);
}
// 三个数字都不同的情况
for (i=0; i<(int)nums.size()-1; i++) {
for (j=i+1; j<nums.size(); j++) {
sum = nums.at(i) + nums.at(j);
if (-sum > nums.at(j) &&
num_cnt.find(-sum) != num_cnt.end()) {
item = getVec(-sum, nums.at(i), nums.at(j));
result.push_back(item);
}
}
}
// 两个数字相同,一个数字不同 或者三个数字均为0
for (map<int, int>::iterator it = num_cnt.begin(); it != num_cnt.end(); it++) {
if (it->first == 0 && it->second >= 3) {
item = getVec(0, 0, 0);
result.push_back(item);
}
else if (it->first != 0 && it->second >= 2) {
sum = 2*it->first;
if (num_cnt.find(-sum) != num_cnt.end()) {
item = getVec(-sum, it->first, it->first);
result.push_back(item);
}
}
}
return result;
}
其实,更好的思想是对vector进行排序,然后固定第一个数,假设为A[i],接下来,我们在A[i]之后的元素中,找到两个数,使其和为-A[i]。 因为数组已经有序,所以可以通过双指针向中间进行夹逼操作。这样我们就把找出三个其和为0的数的问题,转化为在有序数组中,找到两个数其和为某个值的问题。代码如下:
vector<vector<int>> threeSum(vector<int>& nums) {
vector<vector<int>> result;
if (nums.size() < 3)
return result;
// sort vector
sort(nums.begin(), nums.end());
for (int i = 0; i < nums.size() - 2; i++) {
// //avoid duplicate solutions
if (i == 0 || nums[i] > nums[i - 1]) {
int x = -nums[i];
int start = i + 1;
int end = (int)nums.size() - 1;
while (start < end) {
//case 1
if (nums[start] + nums[end] == x) {
vector<int> temp;
temp.push_back(nums[i]);
temp.push_back(nums[start]);
temp.push_back(nums[end]);
result.push_back(temp);
start++;
end--;
//avoid duplicate solutions
while (start < end && nums[end] == nums[end + 1])
end--;
while (start < end && nums[start] == nums[start - 1])
start++;
//case 2
} else if (nums[start] + nums[end] < x) {
start++;
//case 3
} else {
end--;
}
}
}
}
return result;
}<span style="color:#ff0000;">
</span>