题目:740. 删除并获得点数
思路:选了元素x后,所有的x都需要选上,而x-1,x+1都不能选。使用哈希表将nums的每个元素转换为值域数组,然后使用动态规划dp。时间复杂度0(n),细节看注释。
C++版本:
class Solution {
public:
int deleteAndEarn(vector<int>& nums) {
//找到最大值,确定值域数组的范围
int mx=ranges::max(nums);
//值域数组,v[i]表示:所有元素为i的和
vector<int> v(mx+1,0);
for(auto x:nums){
v[x]+=x;
}
//动态规划dp
//f[i]:表示在0~i之间能获得的最大点数
vector<int> f(mx+1,0);
f[1]=v[1];
int ans=f[1];
for(int i=2;i<=mx;i++){
f[i]=max(f[i-1],f[i-2]+v[i]);
ans=max(ans,f[i]);
}
return ans;
}
};
JAVA版本:
class Solution {
public int deleteAndEarn(int[] nums) {
int mx=0;
for(var x:nums){
mx=Math.max(mx,x);
}
int[] v=new int[mx+1];
for(var x:nums){
v[x]+=x;
}
int[] f=new int[mx+1];
f[1]=v[1];
int ans=f[1];
for(int i=2;i<=mx;i++){
f[i]=Math.max(f[i-1],f[i-2]+v[i]);
ans=Math.max(ans,f[i]);
}
return ans;
}
}
Go版本:
func deleteAndEarn(nums []int) int {
mx:=slices.Max(nums)
v:=make([]int,mx+1)
for _,x:=range nums {
v[x]+=x
}
f:=make([]int,mx+1)
f[1]=v[1]
ans:=f[1]
for i:=2;i<=mx;i++ {
f[i]=max(f[i-1],f[i-2]+v[i])
ans=max(ans,f[i])
}
return ans
}