Codeforces Round 901 div2 C + D

C. Jellyfish and Green Apple

题意

有 $n$ 个均为 $1kg$ 的苹果，要平均分给 $m$ 个人，每次操作可以将一块苹果平分成两份。问最少需要多少次操作才能平均分配？（或者$-1$表示无法平均分配）

思路

对于 $n>m$ 的时候，我们只需要给每个人都分一个苹果，直到 $n<m$。所以我们只需考虑 $n<m$ 的情况
每个人可以分到 $\frac{n}{m}$ 的苹果，我们先将其约分成最简分数的形式，也就是分子分母同时除 $gcd(n,m)$，此时如果发现分母 $\frac{m}{gcd(n,m)}$ 不是 $2$ 的整数次幂，那么答案很显然就是 $-1$。否则我们可以使用 $\frac{n}{gcd(n,m)}$ 个 $\frac{m}{gcd(n,m)}$ 来构成小数部分。

判断一个数是否是 $2$ 的整数次幂可以使用：__builtin_popcount()

可以注意到，每切一次，苹果块的数量就会增加 $1$。所以切的次数最少 $\iff$ 最后苹果块的数量最少。

可以证明对于 $\frac{n}{gcd(n,m)}$ 个 $\frac{m}{gcd(n,m)}$，一定是以 $\frac{n}{gcd(n,m)}$ 的二进制形式来构成是最优的。比如 $n=3,m=4$，$n=[11{]}_2$。那么一定是 $2个0.25=0.5$ + $1个0.25$ 来构成小数部分是最优的。

那么最后除了一开始的不用切割的整个完整苹果的分配，小数部分最后每个人都有若干个苹果块的话，减去一开始用来切割成小数部分的 $n$ 个，就是增加的苹果块数量，也即是切割数字了。

// Problem: C. Jellyfish and Green Apple
// Contest: Codeforces - Codeforces Round 901 (Div. 2)
// URL: https://codeforces.com/contest/1875/problem/C
// Memory Limit: 256 MB
// Time Limit: 1000 ms
// 
// Powered by CP Editor (https://cpeditor.org)

#include<bits/stdc++.h>
#define fore(i,l,r)	for(int i=(int)(l);i<(int)(r);++i)
#define fi first
#define se second
#define endl '\n' 

const int INF=0x3f3f3f3f;
const long long INFLL=0x3f3f3f3f3f3f3f3fLL;

typedef long long ll;

int main(){
    std::ios::sync_with_stdio(false);
    std::cin.tie(nullptr);
    std::cout.tie(nullptr);
    int t;
    std::cin>>t;
    while(t--){
    	int n,m;
    	std::cin>>n>>m;
    	n %= m;
    	int a = n / std::__gcd(n,m);
    	int b = m / std::__gcd(n,m);
    	if(__builtin_popcount(b) > 1) std::cout<<-1<<endl;
    	else std::cout<<1ll * __builtin_popcount(a)*m - n<<endl;
    }
	return 0; 
}

---

D. Jellyfish and Mex

题意

有一个长度为 $n$ 的非负整数数组 $a$，$m$ 初始为 $0$
做下列操作 $n$ 次：

• 删除 $a$ 中的一个元素
• 将 $MEX(a)$ 加到 $m$ 上

求最后 $m$ 的最小值

思路

任何长度为 $n$ 的非负整数数组 $a$，其$MEX(a)$ 都 $\le n$

对于当前的 $MEX(a)$，我们一定是选择一个小于 $MEX(a)$ 的 $x$ 来删除，否则就是无用功。一旦选择了一个 $x$ ，剩下的操作就一定是优先将 $x$ 全部删除，从而将 $MEX(a)$ 更新为 $x$。对于那些大于 $x$ 的元素，我们可以留到 $MEX(a)$ 变为 $0$ 后再删除。那么现在要考虑的就是 $x$ 的选择顺序问题，可以肯定的是：$x$ 的选择一定是递减的。

考虑 $DP$：定义 $dp[i]$ 代表在 $MEX(a)=i$ 并且还没有删除任何 $x<i$ 的元素的条件下，表示的 $m$ 的最小值。那么转移方程为：对于某个值 $j$，$\forall i>j,dp[j]=min(dp[j],dp[i]+i\cdot (cnt[j]-1)+j)$，意思就是将 $cnt[j]$ 个 $j$ 全部删除，除了最后一次删除后 $MEX(a)$ 变为 $j$，前面全部的操作 $MEX(a)$ 都是 $i$

// Problem: D. Jellyfish and Mex
// Contest: Codeforces - Codeforces Round 901 (Div. 2)
// URL: https://codeforces.com/contest/1875/problem/D
// Memory Limit: 256 MB
// Time Limit: 1000 ms
// 
// Powered by CP Editor (https://cpeditor.org)

#include<bits/stdc++.h>
#define fore(i,l,r)	for(int i=(int)(l);i<(int)(r);++i)
#define fi first
#define se second
#define endl '\n' 

const int INF=0x3f3f3f3f;
const long long INFLL=0x3f3f3f3f3f3f3f3fLL;

typedef long long ll;

int main(){
    std::ios::sync_with_stdio(false);
    std::cin.tie(nullptr);
    std::cout.tie(nullptr);
    int t;
    std::cin>>t;
    while(t--){
    	int n;
    	std::cin>>n;
    	std::vector<ll> cnt(n+5,0);
    	int x;
    	fore(i,1,n+1){
    		std::cin>>x;
    		if(x <= n)	++cnt[x];
    	}
    	int mex = 0;
    	while(cnt[mex])	 ++mex;
    	std::vector<ll> dp(n+5,INFLL);
    	dp[mex] = 0;
    	for(int i=mex;i>=1;--i)
    		for(int j=0;j<i;++j)
    			dp[j] = std::min(dp[j],dp[i] + 1ll*i*(cnt[j]-1) + j);
    	std::cout<<dp[0]<<endl;
    }
	return 0; 
}