题解 | 田忌赛马

田忌赛马

https://www.nowcoder.com/practice/49d799f65a0749588e9cd7e6135a4a9a

田忌赛马的关键解题思路:

数组arr1元素及排列顺序已定,数组arr2元素已知且可以自由排列。只要arr2数组中存在一种排列顺序能够满足其中任意两元素大于对应arr1中的元素即可win。因此该题的核心点可简化为:求数组arr2的全排列

数组的全排列

暴力解法固然好想到,面对元素个数较少时也好写出。但面对元素个数过多时就难了。

数组全排列可以通过回溯法生成。

简单思考一下,以[1,2,3]为例。

选定第一个元素[1]。选择第二个元素[1,2],[1,3]。选定第三个元素,得到两种排列结果:[1,2,3],[1,3,2]。

重新选定第一个元素[2]。选择第二个元素[2,1],[2,3]。选定第三个元素,得到:[2,1,3],[2,3,1]。

重新选定第一个元素[3]。选择第二个元素[3,1],[3,2]。选定第三个元素,得到:[3,1,2],[3,2,1]。

可以看到每个步骤中都先满足结果条件(temp.size()==nums.length())录入结果后,回溯到上一步,重新选择下一个元素。直到数组中的每个元素都被使用过为止。因此回溯法的通用步骤如下:

回溯法步骤

  1. 初始化:used数组全为false,临时列表temp为空, permutations用于收集所有排列。
  2. 如果temp大小等于3,则加入排列集合。否则,遍历0到2,如果used[i]为false,则选择a[i],标记used[i]为true,然后递归,然后回溯。

代码实现:

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.Scanner;

// 注意类名必须为 Main, 不要有任何 package xxx 信息
public class Main {
    public static void main(String[] args) {
        //输入数组
        Scanner in = new Scanner(System.in);
        // 注意 hasNext 和 hasNextLine 的区别
        int[] arr1 = new int[3];
        int[] arr2 = new int[3];
        for (int i = 0; i < 3; i++) arr1[i] = in.nextInt();
        for (int i = 0; i < 3; i++) arr2[i] = in.nextInt();

        //检查田忌是否能赢
        if (canWin(arr1, arr2)) {
            System.out.println("Yes");
        } else {
            System.out.println("No");
        }

        in.close();


    }

    public static boolean canWin(int[] king, int[] tian) {
        //生成田忌马匹的所有排列
        List<List<Integer>> permutations = generatePermutation(tian);

        for (List<Integer> per : permutations) {
            if (canWinWithOrder(king, per)) {
                return true;
            }
        }

        return false;

    }

    //检查特定出场顺序下tian是否能赢
    private static boolean canWinWithOrder(int[] king, List<Integer>tianOrder) {
        int winCount = 0;

        //对三场比赛进行比较
        for (int i = 0; i < 3; i++) {
            if (tianOrder.get(i) > king[i]) {
                winCount++;

            }
        }


        return winCount >= 2;
    }


    //生成数组的所有排列
    private static List<List<Integer>> generatePermutation(int[] nums) {
        List<List<Integer>> result = new ArrayList<>();
        backtrack(result, new ArrayList<>(), nums, new boolean[nums.length]);
        return result;
    }

    //回溯法生成排列
    private static void backtrack(List<List<Integer>> result, List<Integer> temp,
                                  int[] nums, boolean[] used) {
        //回溯结束条件:当排列数组满了之后停止回溯
        if (temp.size() == nums.length) {
            result.add(new ArrayList<>(temp));
            return;

        }

        for (int i = 0; i < nums.length; i++) {
            if (used[i]) continue;

            used[i] = true;
            temp.add(nums[i]);
            backtrack(result, temp, nums, used);
            temp.remove(temp.size() - 1);
            used[i] = false;

        }

    }


}

全部评论

相关推荐

头像
04-27 15:11
已编辑
华东师范大学 算法工程师
暑期实习从2月开始投,面了两个月,流程该挂的都挂完了,腾讯字节一共号称是1.7w个hc,不知道都发给谁了,估计今年秋招要难顶。Timeline米哈游、美团、蚂蚁、微软等公司直接简历挂穿,没进面。携程:3.3&nbsp;投递、测评3.12&nbsp;笔试3.18&nbsp;一面3.25&nbsp;二面4.13&nbsp;ai面(hr面)4.14&nbsp;英语测评4.23&nbsp;offer(已拒)腾讯:2.6&nbsp;测评2.28&nbsp;wxg一面3.5&nbsp;wxg二面(挂)3.11&nbsp;teg一面3.21&nbsp;teg二面(取消)3.31&nbsp;teg一面4.10&nbsp;teg二面(挂)4.21&nbsp;wxg一面4.24&nbsp;wxg二面(挂)字节:1.28&nbsp;aml约面(取消)3.17&nbsp;火山一面(挂)4.8&nbsp;aml一面(挂)4.20&nbsp;抖音data一面(挂)阿里:3.23&nbsp;投递、测评3.28&nbsp;笔试3.31&nbsp;淘天一面4.8&nbsp;钉钉一面4.9&nbsp;淘天二面4.10&nbsp;阿里控股一面4.12&nbsp;钉钉二面(取消)4.15&nbsp;淘天hr面4.16&nbsp;淘天offer(已接)4.21&nbsp;高德一面(取消)4.22&nbsp;淘宝闪购一面(取消)面试最大的感触是,现在撞上ai转型,一堆老业务急着转向,新业务非常不成熟,研究型的组bar非常高根本进不去,业务侧挂着算法的岗位干的都是工程活,面试却又要问算法,另外agent的落地也远没有那么广,绝大多数还是那套写死的系统调一下llm&nbsp;api或者做做rag,其余少部分真的在搭agent的,基本不能在线上服务用什么很智能的模型,现阶段成本太高,进去大概率就是给垃圾模型从工程方面兜底,除了业务场景的应用和数据经验以外,技术方面很难有什么提升。算法岗做不了基模的还是去搜广推好,之前判断失误了完全没投,秋招不知道还进不进得去。
嵌入式的小白:不错啊,淘天也是挺好的,恭喜
我的求职进度条
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务