一、LeetCode:
bytedance-016. 16. 最短移动距离
(1)题目描述
给定一棵 n 个节点树。节点 1 为树的根节点,对于所有其他节点 i,它们的父节点编号为 floor(i/2) (i 除以 2 的整数部分)。在每个节点 i 上有 a[i] 个房间。此外树上所有边均是边长为 1 的无向边。
树上一共有 m 只松鼠,第 j 只松鼠的初始位置为 b[j],它们需要通过树边各自找到一个独立的房间。请为所有松鼠规划一个移动方案,使得所有松鼠的总移动距离最短。
格式:
输入:
– 输入共有三行。
– 第一行包含两个正整数 n 和 m,表示树的节点数和松鼠的个数。
– 第二行包含 n 个自然数,其中第 i 个数表示节点 i 的房间数 a[i]。
– 第三行包含 m 个正整数,其中第 j 个数表示松鼠 j 的初始位置 b[j]。
输出:
– 输出一个数,表示 m 只松鼠各自找到独立房间的最短总移动距离。
示例:
输入:
5 4
0 0 4 1 1
5 4 2 2
输出:4
解释:前两只松鼠不需要移动,后两只松鼠需要经节点 1 移动到节点 3
提示:
对于 30% 的数据,满足 n,m <=100。
对于 50% 的数据,满足 n,m <=1000。
对于所有数据,满足 n,m<=100000,0<=a[i]<=m, 1<=b[j]<=n。
(2)代码实现
import java.util.*;
public class Main {
public static void main(String[] args) {
Scanner sc = new Scanner(System.in);
int n = sc.nextInt();
int m = sc.nextInt();
int[] a = new int[n+1];
int totalHome = 0;
int res = 0;
for (int i = 1; i <= n; i++) {
a[i] = sc.nextInt();
totalHome += a[i];
}
for (int i = 0; i < m; i++) {
int index = sc.nextInt();
a[index]--;
}
int leftHome = totalHome - m;
for (int i = n; i > 1; i--) {
if (a[i]!=0) {
if (leftHome > 0 && a[i] > 0) {
if (leftHome > a[i]) {
leftHome -= a[i];
} else {
a[i/2] += (a[i]-leftHome);
res += (a[i]-leftHome);
a[i] = 0;
}
continue;
}
a[i/2] += a[i];
res += Math.abs(a[i]); //每次只往上走一步
a[i] = 0;
}
}
System.out.println(res);
}
}
二、LeetCode:
1482. 制作 m 束花所需的最少天数
(1)问题描述
给你一个整数数组 bloomDay,以及两个整数 m 和 k 。
现需要制作 m 束花。制作花束时,需要使用花园中 相邻的 k 朵花 。
花园中有 n 朵花,第 i 朵花会在 bloomDay[i] 时盛开,恰好 可以用于 一束 花中。
请你返回从花园中摘 m 束花需要等待的最少的天数。如果不能摘到 m 束花则返回 -1 。
示例 1:
输入:bloomDay = [1,10,3,10,2], m = 3, k = 1
输出:3
解释:让我们一起观察这三天的花开过程,x 表示花开,而 _ 表示花还未开。
现在需要制作 3 束花,每束只需要 1 朵。
1 天后:[x, _, _, _, _] // 只能制作 1 束花
2 天后:[x, _, _, _, x] // 只能制作 2 束花
3 天后:[x, _, x, _, x] // 可以制作 3 束花,答案为 3
示例 2:
输入:bloomDay = [1,10,3,10,2], m = 3, k = 2
输出:-1
解释:要制作 3 束花,每束需要 2 朵花,也就是一共需要 6 朵花。而花园中只有 5 朵花,无法满足制作要求,返回 -1 。
示例 3:
输入:bloomDay = [7,7,7,7,12,7,7], m = 2, k = 3
输出:12
解释:要制作 2 束花,每束需要 3 朵。
花园在 7 天后和 12 天后的情况如下:
7 天后:[x, x, x, x, _, x, x]
可以用前 3 朵盛开的花制作第一束花。但不能使用后 3 朵盛开的花,因为它们不相邻。
12 天后:[x, x, x, x, x, x, x]
显然,我们可以用不同的方式制作两束花。
示例 4:
输入:bloomDay = [1000000000,1000000000], m = 1, k = 1
输出:1000000000
解释:需要等 1000000000 天才能采到花来制作花束
示例 5:
输入:bloomDay = [1,10,2,9,3,8,4,7,5,6], m = 4, k = 2
输出:9
(2)代码实现
class Solution {
public int minDays(int[] bloomDay, int m, int k) {
if(m>bloomDay.length/k) return -1;
int low=Integer.MAX_VALUE,high=0;
int length=bloomDay.length;
for(int i=0;i<length;i++){
low=Math.min(low,bloomDay[i]);
high=Math.max(high,bloomDay[i]);
}
while(low<high){
int days=(high-low)/2+low;
if(canMake(bloomDay,days,m,k)){
high=days;
}else{
low=days+1;
}
}
return low;
}
public boolean canMake(int[] bloomDay,int days,int m,int k){
int bouquets=0;
int flowers=0;
int length=bloomDay.length;
for(int i=0;i<length&&bouquets<m;i++){
if(bloomDay[i]<=days){
flowers++;
if(flowers==k){
bouquets++;
flowers=0;
}
}else{
flowers=0;
}
}
return bouquets>=m;
}
}
三、LeetCode:
剑指 Offer 60. n个骰子的点数
(1)问题描述
把n个骰子扔在地上,所有骰子朝上一面的点数之和为s。输入n,打印出s的所有可能的值出现的概率。
你需要用一个浮点数数组返回答案,其中第 i 个元素代表这 n 个骰子所能掷出的点数集合中第 i 小的那个的概率。
示例 1:
输入: 1
输出: [0.16667,0.16667,0.16667,0.16667,0.16667,0.16667]
示例 2:
输入: 2
输出: [0.02778,0.05556,0.08333,0.11111,0.13889,0.16667,0.13889,0.1
(2)代码实现
class Solution {
public double[] dicesProbability(int n) {
int[][] dp=new int[n+1][6*n+1];
for(int i=1;i<=6;i++) dp[1][i]=1;
for(int i=2;i<=n;i++)
for(int j=i;j<=6*i;j++)
for(int k=1;k<=6&&k<=j;k++)
dp[i][j]+=dp[i-1][j-k];
double[] ans=new double[6*n-n+1];
for(int i=n;i<=6*n;i++)
ans[i-n]=((double)dp[n][i])/(Math.pow(6,n));
return ans;
}
}