【华为OD题库-044】跳房子1-java

题目

跳房子,也叫跳飞机,是---种世界性的儿童游戏,游戏参与者需要分多个回合按顺序跳到第1格直到房子的最后---格。跳房子的过程中,可以向前跳,也可以向后跳。

假设房子的总格数是count,小红每回合可能连续跳的步教都放在数组steps中,请问数组中是否有一种步数的组合,可以让小红两个回合跳到最后---格?如果有,请输出索引和最小的步数组合。

注意:

数组中的步数可以重复,但数组中的元素不能重复使用。提供的数据保证存在满足题目要求的组合,且索引和最小的步数组合是唯一的
输入描述

第一行输入为房子总格数count,它是int整数类型。

第二行输入为每回合可能连续跳的步数,它是int整数数组类型。
输出描述

返回索引和最小的满足要求的步数组合(顺序保持steps中原有顺序)

备注

count ≤ 1000

0 ≤ steps.length ≤ 5000

-100000000 steps ≤ 100000000
示例1:
输入:

7

[1,4,5,2,2]
输出

[5, 2]
示例2:
输入

9

[-1,2,4,9,6]
输出

[-1,9]
说明

此样例有多种组合满足两回合跳到最后,譬如: [-1,9],[2,6],其中[-1,9]的索引和为0+3=3,[2,6]的索和为1+4=5,所以索引和最小的步数组合[-1,9]

思路

同leetcode:1. 两数之和

  1. 双指针

和本系列 跳房子2 思路一样,不赘述

  1. 哈希表

除了用对象记录值和索引的关系,还可以用map。

遍历数组,看target - nums[i]是否存在map中:

如果存在,说明当前值+(target - nums[i])等于target,那么记录此时的(target - nums[i])和nums[i],并通过map计算当前组合的索引和。

如果不存在,以当前值为key,当前值的索引为val存到map中

最后返回索引和最小的组合

题解

1. 双指针

java 复制代码
package hwod;

import java.util.*;

public class JumpHouse1 {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        sc.nextLine();

        String secondLines = sc.nextLine();
        int[] nums = Arrays.stream(secondLines.substring(1, secondLines.length() - 1).split(",")).mapToInt(Integer::parseInt).toArray();
        int[] res = jumpHouse1(nums, n);
        StringBuilder sb = new StringBuilder("[");
        for (int i = 0; i < res.length; i++) {
            if (i != 0) sb.append(",");
            sb.append(res[i]);
        }
        sb.append("]");
        System.out.println(sb.toString());
    }

    private static int[] jumpHouse1(int[] nums, int target) {
        int[] res = new int[2];
        List<House1> list = new ArrayList<>();
        for (int i = 0; i < nums.length; i++) {
            list.add(new House1(i, nums[i]));
        }
        Collections.sort(list);
        int i = 0, j = list.size() - 1;
        int minIdxSum = Integer.MAX_VALUE;
        while (i < j) {
            int sum = list.get(i).getVal() + list.get(j).getVal();
            if (sum < target) {
                while (i < j && list.get(i).getVal() == list.get(++i).getVal()) ;
            } else if (sum > target) {
                while (i < j && list.get(j).getVal() == list.get(--j).getVal()) ;
            } else {
                int idxSum = list.get(i).getIdx() + list.get(j).getIdx();
                if (idxSum < minIdxSum) {
                    minIdxSum = idxSum;
                    List<House1> resHouses = Arrays.asList(list.get(i), list.get(j));
                    resHouses.sort(Comparator.comparingInt(House1::getIdx));
                    res = new int[]{resHouses.get(0).getVal(), resHouses.get(1).getVal()};
                }
                while (i < j && list.get(i).getVal() == list.get(++i).getVal()) ;
                while (i < j && list.get(j).getVal() == list.get(--j).getVal()) ;
            }
        }
        return res;

    }
}

class House1 implements Comparable<House1> {
    private int idx;
    private int val;


    public int getIdx() {
        return idx;
    }

    public void setIdx(int idx) {
        this.idx = idx;
    }

    public int getVal() {
        return val;
    }

    public void setVal(int val) {
        this.val = val;
    }

    public House1(int idx, int val) {
        this.idx = idx;
        this.val = val;
    }

    @Override
    public int compareTo(House1 o) {
        if (this.val != o.val) return this.val - o.val;
        return this.idx - o.idx;
    }
}

2. 哈希表

java 复制代码
package hwod;

import org.omg.PortableInterceptor.INACTIVE;

import java.util.*;

public class JumpHouse1 {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        sc.nextLine();

        String secondLines = sc.nextLine();
        int[] nums = Arrays.stream(secondLines.substring(1, secondLines.length() - 1).split(",")).mapToInt(Integer::parseInt).toArray();
        int[] res = jumpHouse1(nums, n);
        StringBuilder sb = new StringBuilder("[");
        for (int i = 0; i < res.length; i++) {
            if (i != 0) sb.append(",");
            sb.append(res[i]);
        }
        sb.append("]");
        System.out.println(sb.toString());
    }

    private static int[] jumpHouse1(int[] nums, int target) {
        Map<Integer, Integer> map = new HashMap<>();
        int[] res = new int[2];
        int minIdxSum = Integer.MAX_VALUE;
        for (int i = 0; i < nums.length; i++) {
            if (map.containsKey(target - nums[i])) {
                int idxSum = i + map.get(target - nums[i]);
                if (idxSum < minIdxSum) {
                    minIdxSum = idxSum;
                    res=new int[]{target - nums[i], nums[i]};
                }
            }
            map.put(nums[i], i);
        }

        return res;
    }
}

补充用例

再提供一组示例验证:
输入:

9

[4,1,3,6,5]
输出:

[4,5]

推荐

如果你对本系列的其他题目感兴趣,可以参考华为OD机试真题及题解(JAVA),查看当前专栏更新的所有题目。

相关推荐
程序员-小李4 分钟前
餐厅下单助手系统(Java+MySQL)
java·开发语言·mysql
开心工作室_kaic8 分钟前
springboot496基于java手机销售网站设计和实现(论文+源码)_kaic
java·开发语言·智能手机
像少年啦飞驰点、10 分钟前
SpringBoot + HttpSession 自定义生成sessionId
java·开发语言
珊珊来吃15 分钟前
EXCEL中给某一列数据加上双引号
java·前端·excel
我曾经是个程序员22 分钟前
使用C#生成一张1G大小的空白图片
java·算法·c#
向阳121826 分钟前
mybatis SqlSessionFactory
java·mybatis
mask哥27 分钟前
算法:LeetCode470_用Rand7()实现Rand10()_java实现
java·开发语言
Suwg20931 分钟前
《手写Mybatis渐进式源码实践》实践笔记(第七章 SQL执行器的创建和使用)
java·数据库·笔记·后端·sql·mybatis·模板方法模式
丁总学Java34 分钟前
优化 invite_codes 表的 SQL 创建语句
java·数据库·sql
抓哇FullStack-Junior37 分钟前
设计模式——适配器模式
java·设计模式·适配器模式