题目为 给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
示例:
给定数组 nums = [-1, 0, 1, 2, -1, -4],
满足要求的三元组集合为:
[
[-1, 0, 1],
[-1, -1, 2]
]
题解 第一种思路就是最简单的暴力算法,三层循环嵌套,提交时会提示执行超时,但代码没有问题,代码如下
public static IList<IList<int>> ThreeSum(int[] nums)
{
List<IList<int>> liRes = new List<IList<int>>();
//Dic为了判断是否存在重复项
Dictionary<string, string> dic = new Dictionary<string, string>();
for (int i = 0; i < nums.Length; i++)
{
for (int j = i + 1; j < nums.Length; j++)
{
for (int w = j + 1; w < nums.Length; w++)
{
if (nums[i] + nums[j] + nums[w] == 0)
{
List<int> li = new List<int>
{
nums[i],nums[j],nums[w]
};
StringBuilder sb = new StringBuilder();
List<int> newLi = li.OrderBy(e => e).ToList();
foreach (var item in newLi)
{
sb.Append(item);
}
if (!dic.ContainsKey(sb.ToString()))
{
dic.Add(sb.ToString(), "");
liRes.Add(li);
}
}
}
}
}
return liRes;
}
之后的代码参考了题解官方精选题解链接
还是从最外的for循环开始,进行判断判断条件,思路为,对集合从小到大排序,两个左侧数与最右侧数相加,如果结果大于0,那么意味着最大的数要变小,那么,最右侧左移一位。如果小于0,则左侧的数右移一位。还需要考虑去重。代码如下
public IList<IList<int>> ThreeSum(int[] nums)
{
List<IList<int>> liRes = new List<IList<int>>();
List<int> li = nums.OrderBy(e => e).ToList();
int AllLen = li.Count();
//总数小于3 直接返回
if (AllLen < 3)
{
return liRes;
}
//排序后第一个最小数大于0 直接返回
if (li[0] > 0)
{
return liRes;
}
for (int i = 0; i < AllLen; i++)
{
if (i > 0 && li[i] == li[i - 1])
{
continue;
}
int L = i + 1;
int R = AllLen - 1;
while (L < R)
{
int sum = li[R] + li[L] + li[i];
if (li[R] + li[L] + li[i] == 0)
{
List<int> liNew = new List<int> {
li[R],li[L],li[i]
};
liRes.Add(liNew);
while (L < R && li[L] == li[L + 1])
{
L++;
}
while (L < R && li[R] == li[R - 1])
{
R--;
}
L++;
R--;
}
else if (sum < 0)
{
L++;
}
else if (sum > 0)
{
R--;
}
}
}
return liRes;
}