首先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;
}
如图:第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;
}