当前位置:   article > 正文

LeetCode---126双周赛_leetcode 126

leetcode 126

题目列表

3079. 求出加密整数的和

3080. 执行操作标记数组中的元素

3081. 替换字符串中的问号使分数最小

3082. 求出所有子序列的能量和

一、求出加密整数的和

按照题目要求,直接模拟即可,代码如下

  1. class Solution {
  2. public:
  3. int sumOfEncryptedInt(vector<int>& nums) {
  4. int n=nums.size(),res=0;
  5. for(auto x:nums){
  6. int s = 0, mx = 0;
  7. while(x){
  8. mx=max(mx,x%10);
  9. s=s*10+1;
  10. x/=10;
  11. }
  12. res+=mx*s;
  13. }
  14. return res;
  15. }
  16. };

二、执行操作标记数组中的元素

题目不难,依旧还是只需要模拟,但是代码量不少,要细心,思路如下:

对于每次查询的操作1:只要判断垓下标是否被标记,然后处理即可

对于每次查询的操作2:要把没有标记过的最小的k个数字标记,如果数字相同则下标小的先标记,很显然要排序(两个维度的排序---首先比较数值,其次比较下标),这里讲一个技巧:我们没必要将数值和下标打包在一起(即用pair)排序,我们可以直接对下标进行排序,具体看代码

如何表示一个数是否被标记?可以额外开一个数组,也可以直接在原数组上修改,将标记过的数记为-1

代码如下

  1. class Solution {
  2. public:
  3. vector<long long> unmarkedSumArray(vector<int>& nums, vector<vector<int>>& queries) {
  4. int n = nums.size(), m = queries.size();
  5. vector<long long> ans(m);
  6. vector<int>idx(n);
  7. long long s = 0;
  8. for(int i=0;i<n;i++) {
  9. idx[i]=i;
  10. s+=nums[i];
  11. }
  12. sort(idx.begin(),idx.end(),[&](int x,int y){
  13. return nums[x]!=nums[y]?nums[x]<nums[y]:x<y;
  14. });
  15. for(int i=0,j=0;i<m;i++){
  16. const auto& v = queries[i];
  17. int index = v[0], k = v[1];
  18. if(nums[index]>=0){
  19. s -= nums[index];
  20. nums[index] = -1;
  21. }
  22. while(k&&j<n){
  23. if(nums[idx[j]]<0){
  24. j++;
  25. continue;
  26. }
  27. s -= nums[idx[j]];
  28. nums[idx[j]]=-1;
  29. j++,k--;
  30. }
  31. ans[i]=s;
  32. }
  33. return ans;
  34. }
  35. };

三、替换字符串中的问号使分数最小

这题是思维题:

首先我们要明白字母出现的顺序并不会影响它们对总分数的贡献(因为字母对分数的贡献仅仅只和该字母出现的次数有关,字母与其他字母之间是相互独立的),也就是说我们只要考虑每个 '?' 填哪个字母即可,根据cost的定义,我们优先考虑之前出现次数少的字母对 '?' 进行填充,当出现次数一样少时,我们优先考虑字典序小的字母,然后对选出的字母进行排序,最后按照 '?' 的位置进行替换即可。

代码如下

  1. class Solution {
  2. public:
  3. string minimizeStringValue(string s) {
  4. int n = s.size();
  5. string tmp;
  6. int cnt[26] = { 0 },c = 0;
  7. for(const auto& e:s){
  8. if(e!='?') cnt[e-'a']++;
  9. else c++;
  10. }
  11. auto cmp=[](const pair<int,int>& x,const pair<int,int>& y)->bool{
  12. return x.first!=y.first ? x.first > y.first : x.second > y.second;
  13. };
  14. priority_queue<pair<int,int>,vector<pair<int,int>>,decltype(cmp)> pq(cmp); //小堆
  15. for(int i=0;i<26;i++)
  16. pq.push({cnt[i],i});
  17. while(c--){
  18. auto [x,ch] = pq.top();
  19. pq.pop();
  20. pq.push({x+1,ch});
  21. tmp += 'a'+ch;
  22. }
  23. sort(tmp.begin(),tmp.end());
  24. for(int i=0,j=0;i<n;i++){
  25. if(s[i]=='?')
  26. s[i]=tmp[j++];
  27. }
  28. return s;
  29. }
  30. };

四、求出所有子序列的能量和

这题找子序列中的子序列,看着很绕,其实就是找和为k的子序列能出现在多少个子序列中,即和为k的子序列做出的贡献,拿示例一举例:和为3的子序列有[1,2]和[3],其中[1,2]在2个子序列中出现,[3]在4个子序列中出现,所以答案为2+4=6。很显然每个和为3的子序列的贡献为2^(n-L),其中n为整个数组的长度,L为子序列的长度。

故答案的表达式为 sum(2^(n-L) * num_K_L) 1<=L<=n,num_K_L表示长为L,和为K的子序列个数

如何求长为L,和为K的子序列的个数?

这是一个背包问题,限制条件有两个:1、长为L  2、和为K

设f[i][L][c]表示前i个数中,长为L,和为c的子序列的个数

1、如果当前的数不在和为c的子序列中,则f[i][L][c]=f[i-1][L][c]

2、如果当前的数在和为c的子序列中,则f[i][L][c]=f[i-1][L-1][c-nums[i]]

所以f[i][L][c]=f[i-1][L][c]+f[i-1][L-1][c-nums[i-1]]

初始化:f[i][0][0]=1,因为长为0,和为0的子序列只能是空,只有一个

代码如下

  1. class Solution {
  2. public:
  3. int sumOfPower(vector<int>& nums, int k) {
  4. int n=nums.size();
  5. const int MOD = 1e9+7;
  6. int f[n+1][n+1][k+1];
  7. memset(f,0,sizeof(f));
  8. //f[i][L][j] = f[i-1][L][j] + f[i-1][L-1][j-nums[i]]
  9. for(int i=0;i<=n;i++)
  10. f[i][0][0]=1;
  11. for(int i=0;i<n;i++){
  12. for(int j=1;j<=k;j++){
  13. for(int L=1;L<=i+1;L++){
  14. f[i+1][L][j] = (f[i][L][j] + (j>=nums[i]?f[i][L-1][j-nums[i]]:0))%MOD;
  15. }
  16. }
  17. }
  18. long long ans = 0, pow2 = 1;
  19. for(int i=n;i>0;i--){
  20. ans = (ans + f[n][i][k]*pow2)%MOD;
  21. pow2 = pow2*2%MOD;
  22. }
  23. return ans%MOD;
  24. }
  25. };
  26. // 优化空间
  27. class Solution {
  28. public:
  29. int sumOfPower(vector<int>& nums, int k) {
  30. int n=nums.size();
  31. const int MOD = 1e9+7;
  32. int f[n+1][k+1];
  33. memset(f,0,sizeof(f));
  34. f[0][0]=1;
  35. for(int i=0;i<n;i++){
  36. for(int j=k;j>=nums[i];j--){
  37. for(int L=1+i;L>0;L--){
  38. f[L][j] = (f[L][j] + f[L-1][j-nums[i]])%MOD;
  39. }
  40. }
  41. }
  42. long long ans = 0, pow2 = 1;
  43. for(int i=n;i>0;i--){
  44. ans = (ans + f[i][k]*pow2)%MOD;
  45. pow2 = pow2*2%MOD;
  46. }
  47. return ans%MOD;
  48. }
  49. };

当然我们也可以根据题目直接定义状态:f[i][j]表示前i个数为数组的,元素和为k的能量值

1、如果nums[i]不在子序列和为k的序列中,那么它有选和不选两种可能,f[i+1][j]=f[i][j]*2

2、如果nums[i]在子序列和为k的序列中,那么它只能被选,f[i+1][j]=f[i][j-nums[i]]

举个例子[1,2,3],要求和为3,假设遍历到 i = 2 ,如果nums[i]=3不在我们想要的子序列中,那么它可以选,也可以不选,即f[i][j] * 2,如果nums[i]=3在我们想要的子序列中,那么它只能被选,即f[i][j-nums[i]]

所以状态转移方程为 f[i+1][j]=f[i][j] * 2+ f[i][j-nums[i]]

代码如下

  1. class Solution {
  2. public:
  3. int sumOfPower(vector<int>& nums, int k) {
  4. const int MOD=1e9+7;
  5. int n=nums.size();
  6. vector<vector<long long>>f(n+1,vector<long long>(k+1));
  7. f[0][0]=1;
  8. for(int i=0;i<n;i++){
  9. for(int j=0;j<=k;j++){
  10. f[i+1][j]=(f[i][j]*2+(j>=nums[i]?f[i][j-nums[i]]:0))%MOD;
  11. }
  12. }
  13. return f[n][k];
  14. }
  15. };
  16. //优化空间
  17. class Solution {
  18. public:
  19. int sumOfPower(vector<int>& nums, int k) {
  20. const int MOD=1e9+7;
  21. int n=nums.size();
  22. vector<long long>f(k+1);
  23. f[0]=1;
  24. for(int i=0;i<n;i++){
  25. for(int j=k;j>=0;j--){
  26. f[j]=(f[j]*2+(j>=nums[i]?f[j-nums[i]]:0))%MOD;
  27. }
  28. }
  29. return f[k];
  30. }
  31. };
声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:【wpsshop博客】
推荐阅读
相关标签
  

闽ICP备14008679号