AcWing第60场周赛＋Leetcode第302场周赛

AcWing第60场周赛

第二题：4495.数组操作

分析：a为正整数数组，每次需要输出最小非零元素，可以将数组从小到大先进行排序，每次操作都是要减去最小的非零元素，第一个最小非0元素一次操作之后就变成了0，下一次操作要输出的就是第二个元素a[1]=a[1]-a[0]，依此类推每次都是a[i]-a[i-1]。

代码：

#include <iostream>
#include <algorithm>

using namespace std;

const int N = 100010;

int n, k;
int a[N];

int main()
{
    scanf("%d%d", &n, &k);
    for (int i = 0; i < n; i ++ ) scanf("%d", &a[i]);
    sort(a, a + n);
    n = unique(a, a + n) - a;//unique函数去重，去除大小相等的元素，返回值是去重后的尾地址

    for (int i = 0; i < k; i ++ )
        if (i < n) printf("%d\n", a[i] - a[i - 1]);
        else printf("%d\n", 0);

    return 0;
}

第三题：4496.吃水果

分析：用一个二维数组f[i][j]表示分水果的方案数，意思是前i个同学拿完水果恰有j个同学和左边的同学水果不一样，第1个同学不考虑在内，第2个同学有m种方案，第i-1个同学与左边不同的方案数为f[ i - 1 ][ j - 1 ]，前i个同学拿完水果恰有j个同学和左边不同分两种情况，①第i个同学与左边不同的方案数为 f[ i - 1 ][ j - 1 ] ×（m-1）②第i个同学和左边相同的方案数 f[ i - 1][ j ] 。所以f[ i ][ j ] = f[ i - 1 ][ j - 1 ] ×（m-1）+ f[ i - 1][ j ].

代码：

#include <iostream>
#include <cstring>
#include <algorithm>

using namespace std;

typedef long long LL;

const int N = 2010;
const int mod = 998244353;

int T;
int n, m, k;
LL f[N][N]; // 前 i 个同学拿完后，恰有j个同学和左边不一样

int main()
{
    cin >> n >> m >> k;
    f[1][0] = m;

    for(int i = 1; i <= n; i ++ )
    {
        f[i][0] = m;
        for(int j = 1; j <= k; j ++ )
        {
            f[i][j] = (f[i - 1][j] + f[i - 1][j - 1] * (m - 1)) % mod;
        }
    }

    cout << f[n][k] << endl;

    return 0;
}

---

Leetcode第302场周赛

第一题：6120.数组能形成多少数对

分析：小白写法，先用sort函数排序，然后数组右移判断是否相等，如果相等代表是一个数对，计数加一。这题还可以用到哈希表，但是我不是很熟练。

代码：

class Solution {
public:
    vector<int> numberOfPairs(vector<int>& nums) {
        int n=nums.size();
        if(n==1) return {0,1};
        sort(nums.begin(),nums.end());
        int cnt=0;
        for(int i=1;i<n;++i){
            if(nums[i-1]==nums[i]){
                ++cnt; ++i;
            }
        }
        return {cnt,n-2*cnt};
        
    }
};

第二题：2342.数位和相等对的最大和

分析：用哈希表记录数位和对应的两个最大的元素，遍历数组中的每个数，对于当前数对应的数位和s，对其中的最大元素进行更新。思路参考

代码：

class Solution {
public:
    int maximumSum(vector<int>& nums) {
        // 计算数位和
        auto sum = [](int x) {
            int cnt = 0;
            while (x) {
                cnt += x % 10;
                x /= 10;
            }
            return cnt;
        };
        // pair分别记录最大的两个数，first第一大，second第二大
        unordered_map<int, pair<int, int>> mp;
        int res = -1;
        for (int x: nums) {
            int s = sum(x);
            int& first = mp[s].first;   // 注意是引用
            int& second = mp[s].second;
            if (x > first) {
                // 更新最大元素，原来的最大变为第二大
                second = first;
                first = x;
            } else if (x > second) {
                second = x;
            }
            // 如果second不为0，则first肯定也不为0
            if (second != 0) {
                res = max(res, first + second);
            }
        }
        return res;
    }
};