N数和

Leetcode 刷题日记–N数和

---

前言

最近在准备MS的面试，在题库里刷到这道题，总结一下n数和。
相对比较统一的模板。

---

一、n数和问题

给你一个包含 n 个整数的数组 nums，判断 nums 中是否存在三（二，四）个元素 a，b，c ，使得 a + b + c = 0 ？请你找出所有和为 0 且不重复的n元组。

注意：答案中不可以包含重复的n元组。

二、代码

使用了相对统一的模板解决n数和问题。

#两数和
class Solution(object):
    def twoSum(self, nums, target):
        #res=[];
        for i in range(len(nums)-1):
            if target-nums[i] in nums[i+1:len(nums)]:
                return [i,nums[i+1:len(nums)].index(target-nums[i])+i+1];
                
class Solution(object):
    def twoSum(self, nums, target):
        dict_j={};
        for i in range(0,len(nums)):
            temp=target-nums[i];
            if temp in dict_j:
                return [i,dict_j[temp]]
            dict_j[nums[i]]=i;

'''
#超时
class Solution(object):
    def threeSum(self, nums):
        res=[];
        for i in range(len(nums)-1):
            for j in range(i+1,len(nums)):
                if -(nums[i]+nums[j]) in nums[j+1:len(nums)]:
                    if sorted([nums[i],nums[j],-(nums[i]+nums[j])]) not in res:
                        res.append( sorted([nums[i],nums[j],-(nums[i]+nums[j])]));
        return res;
'''
class Solution():
    def threeSum(self,nums):
        nums.sort();
        ans=set();
        target=0;
        for i in range(len(nums)-2):
            left=i+1;
            right=len(nums)-1;
            while (right>left):
                temp=nums[i]+nums[left]+nums[right];
                if temp==target:
                    ans.add((nums[i],nums[left],nums[right]));
                    left=left+1;
                    right=right-1;
                elif temp>target:
                    right=right-1;
                elif temp<target:
                    left=left+1;
        rec=[];
        for i in ans:
            rec.append(list(i));
        return rec;
#四数之和
class Solution(object):
    def fourSum(self, nums, target):
        nums.sort();
        ans=set();
        for i in range(len(nums)-3):
            for j in range(i+1,len(nums)-2):
                left=j+1;
                right=len(nums)-1;
                while(right>left):
                    temp=nums[i]+nums[j]+nums[left]+nums[right];
                    if target==temp:
                        ans.add((nums[i],nums[j],nums[left],nums[right]));
                        left=left+1;
                        right=right-1;
                    elif temp>target:
                        right=right-1;
                    elif temp<target:
                        left=left+1;
        rec=[];
        for i in ans:
            rec.append(list(i));
        return rec;

总结

许愿 MS offer, 许愿刷题顺利。后续会把作为23届的研一小白找实习的一些经验写出来。