Educational Codeforces Round 164 (Rated for Div 2)

Educational Codeforces Round 164 (Rated for Div. 2)

链接：Dashboard - Educational Codeforces Round 164 (Rated for Div. 2) - Codeforces

D Colored Balls

考虑一个颜色集合，最优分组法是每次选球最多的两种颜色组合，设其数量最多的颜色球为$A$，总球数为$B$，若$B-A\le A$，则其value为$A$；若$B-A\ge A$，其value值为$\lceil B\rceil$。

则可先对数组$a$排序，枚举所选球数的颜色$i$，只需知道满足前$i-1$种颜色构成的颜色集合总球数$j$对应的方案数即可计算答案，方案数的计算则可用dp解决。

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;

const int MAXN = 5000+5;
const int N = 5000;
const ll mod = 998244353;
int n;
int a[MAXN];
ll dp[MAXN][MAXN], ans;

int main(){
    scanf("%d", &n);
    for(int i = 1; i <= n; i++){
        scanf("%d", &a[i]);
    }
    sort(a+1, a+n+1);
    dp[0][0] = 1;
    for(int i = 1; i <= n; i++){
        for(int j = 0; j <= a[i]; j++){
            ans = (ans + dp[i-1][j] * a[i]) % mod;
        }
        for(int j = a[i]+1; j <= N; j++){
            ans = (ans + dp[i-1][j]*((j+a[i]+1)/2)) % mod;
        }
        for(int j = 0; j <= N; j++){
            if(j >= a[i])  dp[i][j] = (dp[i-1][j] + dp[i-1][j-a[i]]) % mod;
            else  dp[i][j] = dp[i-1][j];
        }
    }
    printf("%lld\n", ans);
    return 0;
}

E Chain Reaction

考虑对固定的$k$，计算其答案值。对单个$a_i$，所用的次数为$b_i=\lceil \frac{a_i}{k}\rceil$，如 a = { 1 , 9 , 7 , 6 , 2 , 4 , 7 , 8 , 1 , 3 } a=\{1,9,7,6,2,4,7,8,1,3\} a={1,9,7,6,2,4,7,8,1,3}，$k=3$时， b = { 1 , 3 , 3 , 2 , 1 , 2 , 2 , 3 , 1 , 1 } b=\{1,3,3,2,1,2,2,3,1,1\} b={1,3,3,2,1,2,2,3,1,1}，则$b$数组的每次上升对答案都有相应贡献，则只要考虑每个“波峰”相对上一次“波谷”的增加量即可。补充$b_0=0$，则答案为$3+1+1=5$。

如何对所有$k\in [1,\max (a_i)]$计算这个上升值呢？$b$中的“波峰-波谷”显然也是$a$中的“波峰-波谷”，设其在$a$中的值为$(x,y)$，在某个$k$下对应$b$中的值为$(u,v)$，则对$ans[k]$的贡献为$v-u$。则只要在$ans$数组里一段使$(u,v)$相同的$k$上都加$v-u$，考虑到$x$与$y$是独立的，只要在处理$x$时对每个$k$减去不同的$u$，对每个$y$加上不同的$v$，就可以得到答案。

怎么找到每个$k$对应的区间呢？用上整除分块，上整除分块枚举区间的代码如下（不包含$k=1$的情况，对应区间就是$[n,+\infty )$）：

for(int l = 1, r; l < n; l = r+1){
    int k = (n+l-1)/l;
    r = (n-1)/(k-1);
    printf("%d %d %d\n", l, r, k);
}

完整代码如下，对于区间$[x,y)$要另外单独处理$-1$。

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;

const int MAXN = 100000+5;
int n, A;
int a[MAXN];
ll b[MAXN], c[MAXN];

void add(int l, int r, int d){
    b[l] += d;
    b[r+1] -= d;
}

void solve(int x, int kind){
    for(int l = 1, r; l < x; l = r+1){
        int k = (x+l-1)/l;
        r = (x-1)/(k-1);
        add(l, r, k*kind);
    }
}

int main(){
    scanf("%d", &n);
    for(int i = 1; i <= n; i++){
        scanf("%d", &a[i]);
        A = max(A, a[i]);
    }
    int lst = 0;
    for(int i = 1; i <= n; i++){
        if(a[i] > a[i+1] && a[i] >= a[i-1]){
            solve(lst, -1);
            solve(a[i], 1);
            add(lst, a[i]-1, -1);
            lst = a[i];
        }
        else{
            lst = min(lst, a[i]);
        }
    }
    for(int i = 1; i <= A; i++){
        c[i] = c[i-1] + b[i];
        printf("%lld ", c[i]);
    }
    return 0;
}