Codeforces Round #652 (Div. 2) C,D

首先w[i]==1的情况肯定直接分配最大值。

对于剩余情况：

我如果我们预先把1-n的a[i]值都加上记为sm。

对于w[i]==x的情况：相当于：

选取一段区间[l,r]  r-l+1=x,  

sm的值减去[l+1,r-1]区间内的a[i]。

我们的目的是让所有w[i]对应的区间，减去的值和最小。

显然，让w[i]最大的区间放在最前面，然后依次放下去，减的数和最小。否则只要变东任意一个区间，整体减去的区间会变大。

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
#define ls (o<<1)
#define rs (o<<1|1)
#define pb push_back
const double PI= acos(-1.0);
const int M = 1e5+7;
/*
int head[M],cnt=1;
void init(){cnt=1,memset(head,0,sizeof(head));}
struct EDGE{int to,nxt,w;}ee[M*2];
void add(int x,int y,int w){ee[++cnt].nxt=head[x],ee[cnt].w=w,ee[cnt].to=y,head[x]=cnt;}
*/
ll a[M],w[M];
int main()
{
	ios::sync_with_stdio(false);
  	cin.tie(0);
  	int t;
  	cin>>t;
  	while(t--)
  	{
  		int n,k;
  		cin>>n>>k;
  		ll sm=0;
  		for(int i=1;i<=n;i++)cin>>a[i],sm+=a[i];
  		for(int i=1;i<=k;i++)cin>>w[i];
  		sort(a+1,a+1+n);
  		sort(w+1,w+1+k);
  		int id=1,ed=n;
  		for(int i=k;i>=1;i--)
  		{
  			if(w[i]==1)sm+=a[ed],ed--;
  			else
  			{
  				if(w[i]==2)continue;
  				int c=w[i]-2;
  				for(int j=id+1;j<=id+c;j++)sm-=a[i];
  				id=id+c+1;
			}
		}
		cout<<sm<<endl;
	}
	return 0;
}

 

D - TediousLee

如图：第i层的树，根root有三颗子树，其中2个是第i-2层的树，1个是第i-1层的树。

令dp[i]为第i层树最多能涂色几个节点。r[i]为第i层数最多涂节点的情况下，根节点是否必须选择。

显然：dp[i]=dp[i-1]+dp[i-2]*2+(r[i-1]==0&&r[i-2]==0)*4;    r[i]=(r[i-1]==0&&r[i-2]==0)

然后直接遍历转移即可。

或者你也可以直接令dp[i][2],表示第i层树选根节点/不选根节点  最多涂色节点

 

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
#define ls (o<<1)
#define rs (o<<1|1)
#define pb push_back
const double PI= acos(-1.0);
const int M = 2e6+7;
/*
int head[M],cnt=1;
void init(){cnt=1,memset(head,0,sizeof(head));}
struct EDGE{int to,nxt,w;}ee[M*2];
void add(int x,int y,int w){ee[++cnt].nxt=head[x],ee[cnt].w=w,ee[cnt].to=y,head[x]=cnt;}
*/
const int mod= 1e9+7;
ll dp[M];
int r[M];
int main()
{
	ios::sync_with_stdio(false);
  	cin.tie(0);
  	int t;
  	cin>>t;
	dp[3]=4,r[3]=1;
	dp[4]=4,r[4]=0;
	for(int i=5;i<=2e6;i++)
	{
		int tp=0;
		if(r[i-2]==0&&r[i-1]==0)r[i]=1,tp=4;
		dp[i]=(dp[i-2]*2+dp[i-1]+tp)%mod;
	}
  	while(t--)
  	{
  		int n;
  		cin>>n;
  		cout<<dp[n]<<endl;
	  
	  }
	return 0;
}