提交说明:更新了LeetCode337、LeetCode739、HS、LeetCode705的代码,并添加了测试用例。优化了代码结构,提高了代码的可读性和可维护性。

This commit is contained in:
whaifree 2024-04-14 22:16:36 +08:00
parent 7ed697d929
commit 441579c86c
4 changed files with 176 additions and 5 deletions

View File

@ -49,12 +49,9 @@ public class HS {
public static void main(String[] args) { public static void main(String[] args) {
double[] doubles = {3, 2, 6, 5, 1, 3}; double[] doubles = {3, 2, 6, 5, 1, 3};
System.out.println(get_max_profit(10000, 6, doubles, 2)); System.out.println(get_max_profit(10000, 6, doubles, 2));
} }
/** /**
* 代码中的类名方法名参数名已经指定请勿修改直接返回方法规定的值即可
*
* 根据输入计算最大收益 * 根据输入计算最大收益
* @param M double浮点型 初始资金 * @param M double浮点型 初始资金
* @param N int整型 历史价格天数 * @param N int整型 历史价格天数
@ -62,7 +59,7 @@ public class HS {
* @param K int整型 最大允许交易次数 * @param K int整型 最大允许交易次数
* @return double浮点型 * @return double浮点型
*/ */
public static double get_max_profit (double M, int N, double[] historyPrices, int K) { public static double get_max_profit1 (double M, int N, double[] historyPrices, int K) {
// write code here // write code here
// dp[0] 表示当天手里没有股票的手里的钱 // dp[0] 表示当天手里没有股票的手里的钱
// - 没有钱有两种 // - 没有钱有两种
@ -94,4 +91,72 @@ public class HS {
return res; return res;
} }
/**
* 根据输入计算最大收益
* @param M double浮点型 初始资金
* @param N int整型 历史价格天数
* @param historyPrices double浮点型一维数组 N天历史价格
* @param K int整型 最大允许交易次数
* @return double浮点型
*/
public static double get_max_profit (double M, int N, double[] historyPrices, int K) {
/**
* dp[i][j] 表示第i天手里持有的钱 j为1-2K分别奇数时表示未持有 偶数时表示持有
* 初始化dp[0][持有\偶数] = M/history[i] 表示di 第0天持有的股票数量
*
*/
// 表示手里有股票时的手里有的钱
double[][] buy = new double[N][K];
// 表示手里没有股票时的手里的钱, sellij表示第i天手里有已经没有第j支股票
double[][] sell = new double[N][K];
for (int i = 0; i < K; i++) {
sell[0][i] = M;
}
/**
* 未持有
* - 前一天就未持有
* - 当天卖出
* 持有
* - 前一天就持有
* - 当天买入
*/
for (int i = 1; i < N; i++) {
// 当天有股票
buy[i][0] = Math.max(buy[i - 1][0], -historyPrices[i]);
// 当天没有股票
sell[i][0] = Math.max(sell[i - 1][0], buy[i - 1][0] + buy[i - 1][0] * historyPrices[i]);
for (int j = 1; j < K; j++) {
buy[i][j] = Math.max(buy[i - 1][j], sell[i - 1][j] - historyPrices[i]);
sell[i][j] = Math.max(sell[i - 1][j], buy[i - 1][j] + buy[i - 1][j] * historyPrices[i]);
}
}
return 1;
}
public static int maxProfit(int M, int N, int K, int[] historyPrices) {
int[][] buy = new int[N][K + 1];
int[][] sell = new int[N][K + 1];
buy[0][0] = -M;
for (int i = 1; i < N; i++) {
buy[i][0] = Math.max(buy[i-1][0], -historyPrices[i]);
sell[i][0] = Math.max(sell[i-1][0], buy[i-1][0] + historyPrices[i]);
for (int j = 1; j <= K; j++) {
buy[i][j] = Math.max(buy[i-1][j], sell[i-1][j-1] - historyPrices[i]);
sell[i][j] = Math.max(sell[i-1][j], buy[i-1][j] + historyPrices[i]);
}
}
return sell[N-1][K];
}
} }

View File

@ -0,0 +1,67 @@
package cn.whaifree.leetCode.Hash;
import org.junit.Test;
/**
* @version 1.0
* @Author whai文海
* @Date 2024/4/14 13:15
* @注释
*/
public class LeetCode705 {
@Test
public void test()
{
MyHashSet myHashSet = new MyHashSet();
myHashSet.add(1); // set = [1]
myHashSet.add(2); // set = [1, 2]
System.out.println(myHashSet.contains(1)); // 返回 True
System.out.println(myHashSet.contains(3)); // 返回 False 未找到
System.out.println(myHashSet.contains(2));
myHashSet.remove(2);
System.out.println(myHashSet.contains(2)); // 返回 False 已移除
}
/**
* int 表示4字节 32位 每一位代表一个位置
* 1000000/32 40000表示桶
*/
class MyHashSet {
int[] bucket = null;
public MyHashSet() {
bucket = new int[40000];
}
public void add(int key) {
int bkLoc = key / 32;
int ex = key % 32;
int num = bucket[bkLoc];
// 如num=4 (100) 1<<1= 010 进行|操作后变成 110
bucket[bkLoc] = num | (1 << ex);
}
public void remove(int key) {
int bkLoc = key / 32;
int ex = key % 32;
int num = bucket[bkLoc];
// 如num=7 (111) 1<<1= 010 进行~后变成101 & 操作后变成 101
bucket[bkLoc] = num & ~(1 << ex);
}
public boolean contains(int key) {
int bkLoc = key / 32;
int ex = key % 32; // 偏移
int num = bucket[bkLoc]; // 数字
return ((num >> ex) & 1) == 1;
// 如num=7 (111) 1<<1= 010 进行~后变成101 & 操作后变成 101
// int i = num & (1 << ex);
// return 1 << ex == i;
}
}
}

View File

@ -1,4 +1,4 @@
package cn.whaifree.MonotoneStack; package cn.whaifree.leetCode.MonotoneStack;
import org.junit.Test; import org.junit.Test;

View File

@ -1,5 +1,8 @@
package cn.whaifree.redo.redo_24_4_13; package cn.whaifree.redo.redo_24_4_13;
import cn.whaifree.leetCode.model.TreeNode;
import org.junit.Test;
/** /**
* @version 1.0 * @version 1.0
* @Author whai文海 * @Author whai文海
@ -7,4 +10,40 @@ package cn.whaifree.redo.redo_24_4_13;
* @注释 * @注释
*/ */
public class LeetCode337 { public class LeetCode337 {
@Test
public void test() {
Solution solution = new Solution();
System.out.println(solution.rob(TreeNode.constructTreeByArray(3,2,3,null,3,null,1)));
}
class Solution {
/**
* 0 表示不选该点最大收益
* 1 表示选择该点的最大收益
* @param root
* @return
*/
public int rob(TreeNode root) {
int[] res = dpIn(root);
return Math.max(res[0], res[1]);
}
public int[] dpIn(TreeNode root) {
if (root == null) {
return new int[]{0, 0};
}
int[] dp = new int[2];
int[] left = dpIn(root.left);
int[] right = dpIn(root.right);
// 不选择本点,那么子节点随便选择只要收益大
dp[0] = Math.max(left[0], left[1]) + Math.max(right[0], right[1]);
// 选择了本点子节点就不能选
dp[1] = left[0] + right[0] + root.val;
return dp;
}
}
} }