百度算法题

问题描述

小度的特种部队一共有n名士兵, 一天小度派所有士兵去探索野区。士兵们出发时沿着一条道路行进, 直到遇到三岔路口。
小度在出发前就给部队部署了部队划分规则: 当遇到三岔路口的时候, 部队若可以分为两个部分，并且两个部分的人数差恰好为k, 那么就完成部队划分, 划分的两个部分分别沿着两条路行进下去，否则该部队的所有士兵就在此位置停下扎营。
野区内有不计其数的三岔路口, 所以整个部队的每一个部分最终都会停下扎营，小度想知道最终扎营的总数为多少？

输入描述:
包括两个整数, 即部队中士兵的总人数和划分部队的参数。

输出描述:
一个整数，表示最终答案。

输入例子1:
10 2

输出例子1:
5

例子说明1:
10
/
6 4
/\ /
4 2 3 1
/
3 1
最终叶子节点个数即答案:5

import java.util.LinkedList;
import java.util.Queue;
import java.util.Scanner;
public class Soldiers {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        int m = sc.nextInt();
        Queue<Integer> queue = new LinkedList<>();
        queue.add(n);
        int res =1;
        while (!queue.isEmpty()) {
            Queue<Integer> temp = new LinkedList<>();
            int l = queue.size();
            for (int i = 0; i < l; i++) {
                temp.add(queue.peek());
                if (!queue.isEmpty()&&queue.peek() > m + 1&&(queue.peek()-m)%2==0) {
                    int p = (queue.poll()-m)/2;
                    queue.add(p);
                    queue.add(p + m);
                    res++;
                }else {
                    queue.remove();
                }
            }
            if (temp.equals(queue)) {
                break;
            }
        }
        System.out.println(res);
    }
}

---

解决方案：

队列层序遍历解决