算法:两数之和(Java语言)

前言

Hi,我是柚子,这是一道可爱的算法题~


问题描述

给定一个整数数组 nums 和一个整数目标值 target,请你在该数组中找出和为目标值 target 的那两个整数,并返回它们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素在答案里不能重复出现。

示例 1:
输入:nums = [2,7,11,15], target = 9
输出:[0,1]
解释:因为 nums[0] + nums[1] == 9 ,返回 [0, 1] 。

提示:
2 <= nums.length <= 104
-109 <= nums[i] <= 109
-109 <= target <= 109
只会存在一个有效答案


解决方案

1.方案一

双重for循环 + HashSet

public int[] twoSum1(int[] nums, int target) {
        int length = nums.length;
        Set<Integer> correctSet = new HashSet<>();
        for (int i = 0; i < length; i++) {
            for (int j = 0; j < length; j++) {
                if (i != j) {
                    if (nums[i] + nums[j] == target) {
                        correctSet.add(i);
                        correctSet.add(j);
                    }
                }
            }
        }
        Integer[] correctSum = correctSet.toArray(new Integer[correctSet.size()]);
        return Arrays.stream(correctSum).mapToInt(Integer::valueOf).toArray();
    }

在这里插入图片描述

2.方案二

双重for循环 + 数组

public int[] twoSum2(int[] nums, int target) {
        int length = nums.length;
        int[] ans = new int[2];
        for (int i = 0; i < length; i++) {
            for (int j = 0; j < length; j++) {
                if (i != j) {
                    if (nums[i] + nums[j] == target) {
                        ans[0] = i;
                        ans[1] = j;
                        return ans;
                    }
                }
            }
        }
        return new int[0];
    }

在这里插入图片描述

3.方案三

双重for循环(j=i+1)+ 数组

public int[] twoSum3(int[] nums, int target) {
        int[] ans = new int[2];
        int n = nums.length;
        for(int i = 0; i < n; i++){
            for(int j = i + 1; j < n; j++){
                if(nums[i] + nums[j] == target){
                    ans[0] = i;
                    ans[1] = j;
                    return ans;
                }
            }
        }
        return new int[0];
    }

在这里插入图片描述

4.方案四

单层for循环 + HashMap

public int[] twoSum4(int[] nums, int target) {
        int n = nums.length;
        HashMap<Integer, Integer> map = new HashMap<>();
        for (int i = 0; i < n; i++) {
            int diff = target - nums[i];
            if (map.containsKey(diff)) {
                return new int[]{map.get(diff), i};
            } else {
                map.put(nums[i], i);
            }
        }
        return new int[0];
    }  

在这里插入图片描述


题目来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/two-sum

╭◜◝ ͡ ◜◝╮
( ˃̶͈◡˂ ̶͈ )感觉有用的话,欢迎点赞评论呀!
╰◟◞ ͜ ◟◞╯

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
Prim算法: ```java import java.util.ArrayList; import java.util.HashSet; import java.util.List; import java.util.PriorityQueue; import java.util.Set; public class PrimMST { public static List<Edge> primMST(Graph graph) { List<Edge> result = new ArrayList<>(); Set<Integer> visited = new HashSet<>(); PriorityQueue<Edge> pq = new PriorityQueue<>((a, b) -> a.weight - b.weight); visited.add(0); for (Edge e : graph.edges[0]) { pq.offer(e); } while (!pq.isEmpty() && visited.size() < graph.vertices) { Edge e = pq.poll(); if (visited.contains(e.to)) { continue; } visited.add(e.to); result.add(e); for (Edge next : graph.edges[e.to]) { if (!visited.contains(next.to)) { pq.offer(next); } } } return result; } } ``` Kruskal算法: ```java import java.util.ArrayList; import java.util.Collections; import java.util.Comparator; import java.util.List; public class KruskalMST { public static List<Edge> kruskalMST(Graph graph) { List<Edge> result = new ArrayList<>(); UnionFind uf = new UnionFind(graph.vertices); List<Edge> edges = new ArrayList<>(); for (int i = 0; i < graph.vertices; i++) { edges.addAll(graph.edges[i]); } // sort edges by weight Collections.sort(edges, Comparator.comparingInt(a -> a.weight)); for (Edge e : edges) { int root1 = uf.find(e.from); int root2 = uf.find(e.to); if (root1 != root2) { uf.union(root1, root2); result.add(e); } } return result; } } ``` 测试代码: ```java import org.junit.jupiter.api.Assertions; import org.junit.jupiter.api.Test; import java.util.List; public class TestMST { @Test public void testPrim() { Graph g = new Graph(5); g.addEdge(0, 1, 2); g.addEdge(0, 3, 6); g.addEdge(1, 3, 8); g.addEdge(1, 2, 3); g.addEdge(1, 4, 5); g.addEdge(2, 4, 7); g.addEdge(3, 4, 9); List<Edge> result = PrimMST.primMST(g); Assertions.assertEquals(result.size(), 4); Assertions.assertTrue(result.contains(new Edge(0, 1, 2))); Assertions.assertTrue(result.contains(new Edge(1, 2, 3))); Assertions.assertTrue(result.contains(new Edge(1, 4, 5))); Assertions.assertTrue(result.contains(new Edge(0, 3, 6))); } @Test public void testKruskal() { Graph g = new Graph(5); g.addEdge(0, 1, 2); g.addEdge(0, 3, 6); g.addEdge(1, 3, 8); g.addEdge(1, 2, 3); g.addEdge(1, 4, 5); g.addEdge(2, 4, 7); g.addEdge(3, 4, 9); List<Edge> result = KruskalMST.kruskalMST(g); Assertions.assertEquals(result.size(), 4); Assertions.assertTrue(result.contains(new Edge(0, 1, 2))); Assertions.assertTrue(result.contains(new Edge(1, 2, 3))); Assertions.assertTrue(result.contains(new Edge(1, 4, 5))); Assertions.assertTrue(result.contains(new Edge(0, 3, 6))); } } ```

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

柚子猫屿

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值