Codeforces Round 958 (Div. 2)

A.思维：https://codeforces.com/contest/1988/problem/A

AC代码：

#include<bits/stdc++.h>
using namespace std;
int t;
int main(){
	cin>>t;
	while(t--){
		int n,k;
		cin>>n>>k;
		int cnt;
		if(k>=n){
            if(n==1) cout<<0<<endl; 
			else cout<<1<<endl;
		}
		else{
		if((n-k)%(k-1)==0){
			cnt=(n-k)/(k-1);
		}
		else{
			cnt=(n-k)/(k-1)+1;
		}
		cout<<cnt+1<<endl;
		}

	}
}

B.思维，模拟https://codeforces.com/contest/1988/problem/B

两个点：1.一串0等同于一个0  2.1的个数要比0多1才可以

AC代码：

#include<bits/stdc++.h>
using namespace std;
int t,n,a[200010];
int main(){
    cin>>t;
    while(t--){
        cin>>n;
        for(int i=1;i<=n;i++) cin>>a[i];
        int cnt0=0,cnt1=0;
        for(int i=1;i<=n-1;i++){
            if(a[i]==1) cnt1++;
            else{
                if(a[i]==a[i+1]) continue;
                else cnt0++;
            }
        }
        if(a[n]==1){
            cnt0++;
            cnt1++;
        }
        else{
            cnt0++;
        }
        if(cnt0<=cnt1-1) cout<<"YES"<<endl;
        else cout<<"NO"<<endl;
    }
}

C.数学：https://codeforces.com/contest/1988/problem/C

手搓下样例就可以发现规律：把n看成一个二进制串，每次从小到大扣掉一个1即可。

其实道理也比较简单，比如n=101101：

我们倒着从大到小来：101101一定是第一个，第二个如果粗糙的想就是101100，因为他第二大。

事实上，假如是一个....1的串那么由他得到答案，而我们一定可以在它与n之间插入101100，假如是...0的串，那101100显然更优，因为每一个位置上的1它存在101100也存在。

接着，后面一个一定是...1的形式，我们考虑这样的第3小101001,此时我们还是依据此时变0的位讨论，同样的可以推出101001相比于100101这种更优（我们把此时第二大的当成刚才最大的，也是同样的逻辑），依次类推即可。

下面是AC代码：

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
ll t,a[200010];
ll n;
int main(){
	cin>>t;
	while(t--){
		cin>>n;
		int cnt=0;
		a[++cnt]=n;
		for(int i=0;i<=60;i++){
			ll ck=1;
			if((n>>i&ck)&&(n-(ck<<i))!=0) a[++cnt]=n-(ck<<i);
		}
		reverse(a+1,a+cnt+1);
		cout<<cnt<<endl;
		for(int i=1;i<=cnt;i++) cout<<a[i]<<" ";
		cout<<endl;
	}}

D.树形DP：https://codeforces.com/contest/1988/problem/D

第一眼还以为是没有上司的舞会，后来发现也大差不差，只要把原来选和不选两个维度替换成什么时候选这个问题上就可以了。

事实上怪物可以被全部消灭的轮数是级别。这里我就直接取了23：

AC代码：

#include<bits/stdc++.h>
typedef long long ll;
using namespace std;
ll t,n,a[300010];
vector<int> edge[300010];
ll dp[300010][100];
void dfs(int root,int fa){
	for(int k=1;k<=23;k++) dp[root][k]=k*a[root];
	for(int i=0;i<edge[root].size();i++){
		int x=edge[root][i];
		if(x==fa) continue;
		dfs(x,root);
		for(int k=1;k<=23;k++){
			ll res=1e18;
			for(int w=1;w<=23;w++){
				if(k==w) continue;
				res=min(res,dp[x][w]);
			}
			dp[root][k]+=res;
		}
	}
}
int main(){
	cin>>t;
	while(t--){
		cin>>n;
		memset(edge,0,sizeof(edge));
		for(int i=1;i<=n;i++) cin>>a[i];
		for(int i=1;i<=n-1;i++){
			int u,v;
			scanf("%d%d",&u,&v);
			edge[u].push_back(v);
			edge[v].push_back(u);
		}
		ll ans=1e18;
		dfs(1,-1);
		for(int i=1;i<=23;i++) ans=min(ans,dp[1][i]);
		cout<<ans<<endl;
	}
}