LeetCode第252场周赛

心情

题目在此
第二次打LeetCode，内心平静，写出三道的，还是太菜了，没能写出第四道(DP)

第一题(三除数)

n很小，直接暴力取模

第二题(你可以工作的最大周数)

可以发现，只有数组中最大的那个数，影响最终的结果。
如果最大的那个数小于等于总和的一半，那么就一定可以每天一个任务，把全部做完。
如果最大的那个数大于总和的一半，那么结果就是总和减去最大的那个值乘二。

typedef long long LL;
class Solution {
public:
    LL numberOfWeeks(vector<int>& m) {
        sort(m.begin(),m.end());
        int le = m.size();
        LL sum = 0;
        for (int i = 0; i < le; i ++)sum+=m[i];
        LL res = 0;
        if (m[le-1] <= (sum+1)/2)return sum;
        else return (sum-m[le-1])*2+1;
    }
};

第三题(收集足够苹果的最小花园周长)

就是简单的推出数学公式
边长：每次加2 看图很容易推出出公式，直接给出代码。

typedef long long LL;
class Solution {
public:
    long long minimumPerimeter(long long n) {
        if (n <= 12)return 8;
        LL sum = 12;
        LL i;
        LL cnt = 2;
        for (i = 2; sum < n; i ++){
            cnt += 2;
            LL temp = 12*i*i;
            sum += temp;
            
        }
        return cnt*4;
    }
};

第四题(统计特殊子序列的数目)

题意很简单，就是求子序列个数。
dp[i][k] k取[0,2] 表示第i个数及以前，以[0,2]结尾的子序列有多少个(不一定是第i个以[0,2]结尾,之前都可能)

const int mod = 1e9+7;
const int maxn = 1e5+10;
int dp[maxn][3];
class Solution {
public:
    int countSpecialSubsequences(vector<int>& nums) {
        int n = nums.size();
        for (int i = 1; i <= n; i ++){
            int v = nums[i-1];
            for (int k = 0; k < 3; k ++)dp[i][k] = dp[i-1][k];      //最不济也是上一个继承过来
            if (v==0) dp[i][v] = (dp[i][v]+1)%mod;     //假设以这个0为起始
            dp[i][v] = (dp[i][v]+dp[i-1][v])%mod;     //在前面的v数量上加一
            if (v > 0)dp[i][v] = (dp[i][v]+dp[i-1][v-1])%mod;   //连在上一个v-1上面
        }
        return dp[n][2];
    }
};

优化：思路来自大佬
对于每个数字 k ，状态转移都分三种情况：
不选择当前数字，状态值保持不变 dp[i][k] += dp[i-1][k]
当前数字之前已存在，当前只是重复选择 dp[i][k] += dp[i-1][k]
当前数字之前不存在，第一次选择 dp[i][k] += dp[i-1][k-1]
三种状态合并就是： dp[i][k] = dp[i-1][k] * 2 + dp[i-1][k-1]。

对于状态 0，由于不存在上一个数字，所以第一次选择是答案加 1。
为了保持计算统一，下标可以从 1 开始，所有dp[i][0]的值都设置为 1 即可保持一致。
另外，可以发现，状态 dp[i] 只与 dp[i-1] 有关，所以可以将其去掉可以使用滚动数组甚至一个数组上迭代计算。

int countSpecialSubsequences(vector<int>& nums) {
    ll dp[4] = {1, 0, 0, 0};
    for (int i = 0; i < n; i ++){
        int v = nums[i];
        v++;
        dp[v] = (dp[v] * 2 + dp[v-1]) % mod;
    }
    return dp[3];
}

我只求能写出来，不求优化😭