[数据结构与算法] 随机数索引（java）

开发: C++知识库 Java知识库 JavaScript Python PHP知识库人工智能区块链大数据移动开发嵌入式开发工具数据结构与算法开发测试游戏开发网络协议系统运维
教程: HTML教程 CSS教程 JavaScript教程 Go语言教程 JQuery教程 VUE教程 VUE3教程 Bootstrap教程 SQL数据库教程 C语言教程 C++教程 Java教程 Python教程 Python3教程 C#教程
数码: 电脑笔记本显卡显示器固态硬盘硬盘耳机手机 iphone vivo oppo 小米华为单反装机图拉丁

-> 数据结构与算法 -> 随机数索引（java） -> 正文阅读

[数据结构与算法]随机数索引（java）

问题描述：
在这里插入图片描述
样例如下：

代码如下：利用map解决即可

import java.util.*;
public class Solution {
    //随机数索引
    Map<Integer, List<Integer>> map=new HashMap<>();//利用map来存储nums数组中的值和下标
    public Solution(int[] nums) {
        for (int i = 0; i <nums.length ; i++) {
            if (map.containsKey(nums[i])){//利用map查重，若存在，则直接添加后修改
                List<Integer> list=map.get(nums[i]);
                list.add(i);
                map.put(nums[i],list);
            }else {
                List<Integer> list=new LinkedList<>();
                list.add(i);
                map.put(nums[i],list);
            }
        }
    }
    public int pick(int target) {
     int res=0;
     if (map.containsKey(target)) {
         List<Integer> list = map.get(target);
         if (list.size() == 1) res = list.get(0);
         else {
             int random = new Random().nextInt(list.size());
             res = list.get(random);
         }
         return res;
     }else return -1;
    }

    public static void main(String[] args) {
        Scanner scanner=new Scanner(System.in);
        System.out.println("请输入数组长度");
        int n=scanner.nextInt();
        int[] nums=new int[n];
        System.out.println("请输入数组内容");
        for (int i = 0; i <n ; i++) {
            nums[i]=scanner.nextInt();
        }
        Solution solution=new Solution(nums);
        while (true) {
            System.out.println("请输入要查找的数字");
            int target = scanner.nextInt();
            System.out.println(solution.pick(target));
        }

    }
}