題目什麼的都不貼了,來的人都看過題目,以後這個主要是寫給我自己覆盤總結,順便蹭點粉。
代碼:
class Solution {
public List<List<Integer>> fourSum(int[] nums, int target) {
Arrays.sort(nums);
List<List<Integer>> list = new ArrayList<List<Integer>>();
for(int i=0;i<nums.length-3;i++){
if(i>0&&nums[i]==nums[i-1]) continue;
for(int j=i+1;j<nums.length-2;j++){
if(j>i+1&&nums[j]==nums[j-1]) continue;
int lo = j+1;
int hi = nums.length-1;
while(lo<hi){
if(nums[i]+nums[j]+nums[lo]+nums[hi]==target){
list.add(Arrays.asList(nums[i], nums[j], nums[lo],nums[hi]));
while(lo<hi&&nums[lo]==nums[lo+1]) lo++;
while(lo<hi&&nums[hi]==nums[hi-1]) hi--;
hi--;
lo++;
}
else if(nums[i]+nums[j]+nums[lo]+nums[hi]>target){
hi--;
}
else{
lo++;
}
}
}
}
return list;
}
}
分析: 這個時間複雜度是O(n3),主要處理過程就是,第一防止重複,第二就是注意邊界,i應該不能超過nums.length-3 然後我就想:有三數之和,二數之和,四數之和,那是不是有k數之和?
然後看下討論,居然有人寫了k數之和的遞歸代碼,nb!
k數之和代碼:
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
public class Solution{
int len=0;
public List<List<Integer>> fourSum(int [] nums, int target){
len=nums.length;
Arrays.sort(nums);
return kSum(nums,target,4,0);
}
private ArrayList<List<Integer>> kSum(int[] nums,int target,int k,int index){
ArrayList<List<Integer>> res = new ArrayList<>();
if(index>=len){
return res;
}
if(k==2){
int i=index,j=len-1;
while(i<j){
if(target-nums[i]==nums[j]){
List<Integer> temp = new ArrayList<>();
temp.add(nums[i]);
temp.add(nums[j]);
res.add(temp);
//跳過重複
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){
j--;
}
else{//往右邊移動
i++;
}
}
}
else{
for(int i=index;i<len-k+1;i++){
//跳過重複
if(i>index&&nums[i]==nums[i-1]){
continue;
}
ArrayList<List<Integer>> temp = kSum(nums,target-nums[i],k-1,i+1);
if(temp!=null){
for(List<Integer> t : temp){
t.add(0,nums[i]);
}
res.addAll(temp);
}
}
}
return res;
}
}