力扣 第488场周赛(A~D)

A:3833. 统计主导元素下标数

算一下sum，然后一边遍历一边减即可判断当前元素是否是主导元素。

class Solution {
public:
    int dominantIndices(vector<int>& nums) {
        int sum=accumulate(nums.begin(),nums.end(),0);
        int ans=0,n=nums.size();
        for(int i=0;i<n-1;i++){
            sum-=nums[i];
            if(nums[i]>float(sum)/(n-i-1))
                ans++;
        }
        return ans;
    }
};

B:3834. 合并相邻且相等的元素
要求合并最左侧相邻相等的元素，合并之后放回原位置，可再次合并。分析后，操作完美符合栈。

class Solution {
public:
    vector<long long> mergeAdjacent(vector<int>& nums) {
        vector<long long> S;
        for(int i=0;i<nums.size();i++){
            S.push_back(nums[i]);
            while(S.size()>=2){
                int t=S.size();
                if(S[t-1]!=S[t-2]) break;
                long long a=S[t-1];
                S.pop_back();
                S.pop_back();
                S.push_back(a*2);
            }
        }
        return S;
    }
};

C:3835. 开销小于等于 K 的子数组数目
定义一个数组的cost为(max-min)*len，用单调队列维护min，max。

class Solution {
public:
    long long countSubarrays(vector<int>& nums, long long k) {
        long long n=nums.size();
        long long ans=0;
        deque<long long> qmin,qmax;
        long long l=0,r;
        auto getcost=[&](){
            return (nums[qmax.front()]-nums[qmin.front()])*(r-l+1);
        };
        for(r=0;r<n;r++){
            while(!qmin.empty() && nums[qmin.back()]>=nums[r]) qmin.pop_back();
            qmin.push_back(r);
            while(!qmax.empty() && nums[qmax.back()]<=nums[r]) qmax.pop_back();
            qmax.push_back(r);
            while(getcost()>k){
                if(qmin.front()==l)
                    qmin.pop_front();
                if(qmax.front()==l)
                    qmax.pop_front();
                l++;
            }
            ans+=r-l+1;
        }
        return ans;
    }
};

D:3836. 恰好 K 个下标对的最大得分

给定两个数组，从数组a和数组b中分别选k个元素，得分为这两个k维向量的点积，问最大得分为多少？

定义f(i,j,k)为从a中前i个和b中前j个分别选k个元素的最大得分。

则有初始化k==0---->f=0

　　　　i<k || j<k ------->f=-INF

由于存在负数元素，且求最大值，那么除了上面为0的元素，其他的都可初始化为-INF。

转移方程，考虑a选不选第i个和b选不选第j个。

f(i,j,k)=max(f(i-1,j,k),f(i,j-1,k),f(i-1,j-1,k-1)+ai*bi)

const int N = 110;
typedef long long LL;
LL f[N][N][N];
class Solution {
public:
    long long maxScore(vector<int>& nums1, vector<int>& nums2, int x) {
        const LL INF=1ll<<60;
        for(int i=0;i<N;i++){
            for(int j=0;j<N;j++){
                for(int k=0;k<N;k++){
                    f[i][j][k]=-INF;
                }
            }
        }
        for(int i=0;i<N;i++){
            for(int j=0;j<N;j++){
                f[i][j][0]=0;
            }
        }
        int n=nums1.size();
        int m=nums2.size();
        
        for(int i=1;i<=n;i++){
            for(int j=1;j<=m;j++){
                for(int k=1;k<=x;k++){
                    f[i][j][k]=max(f[i][j-1][k],f[i-1][j][k]);
                    if(f[i-1][j-1][k-1]!=-INF)
                        f[i][j][k]=max(f[i][j][k],f[i-1][j-1][k-1]+1ll*nums1[i-1]*nums2[j-1]);
                }
            }
        }
        return f[n][m][x];
    }
};