第十四届蓝桥杯JavaB组省赛石子合并

0合并石子 - 蓝桥云课 (lanqiao.cn)

典型的区间$dp$，不过和以往不同的是多了只能相同颜色之间合并，并且要求出最少堆数和最小费用。

那我们自然而然的可以把定义：$f_{i,j,k}$表示的是合并$[i,j]$颜色为$k$的最小费用。此时要注意的每次合并都会让颜色变化。所以我们的突破口就是颜色，并且$cn{t}_{i,j}$表示的是区间$[i,j]$最少有几堆石头，$w_{i,j}$表示区间$[i,j]$的最小费用。

假设：$f_{i,j,k}=inf$表示的是从区间$[i,j]$不能合并过去。$[i,j]$颜色都是$k$的话这时候就是裸的区间$dp$,如果$[i,t],[t+1,j]$这两个区间颜色不一样那么就可以分别做区间$dp$。

初始化显然如果是$f_{i,i,k}$一定是$0$，而$cn{t}_{i,j}=j-i+1$。

我们再来处理最小堆数，我们做完合并以后就可以判断一下，如果$f_{i,j,k}=inf$的话说明$f_{i,t,k}\cup f_{t+1,j,k}$ 颜色不一样并不能转移。

如果可以转移的话

$f_{i,j,(k+1)mod3}=min(f_{i,j,(k+1)mod3},f(i,t,k)+f(t+1,j,k)+s_j-s_i)$，$cnt[i][j]=1$,$w_{i,j}=min(f_{i,j,0},f_{i,j,1},f_{i,j,2})$

因为$f_{i,j,k}$最后得出的是相同颜色最小费用，如果有不能合并的话此时的$f_{i,j,k}=+\infty ,cn{t}_{i,j}=+\infty ,w_{i,j}=+\infty$不能保证是最终答案（因为最终答案是总费用最小）所以我们还需要对$cnt$和$w$再做一次区间$dp$，优先把堆数少的先合并，如果堆数一样，那就按照费用最低来合并

#include<bits/stdc++.h>
#define int long long
#define x first
#define y second
using namespace std;
const int N=500,inf=0x3f3f3f3f;
int n;
int f[N][N][5];
int a[N],w[N][N];
int cnt[N][N];
int s[N];
void solve()
{
	cin>>n;
	for(int i=1;i<=n;i++)
	    cin>>a[i];
	for(int i=1;i<=n;i++)
	   s[i]=s[i-1]+a[i];
	for(int i=1;i<=n;i++)
	{
		for(int j=i;j<=n;j++)
		{
			for(int k=0;k<3;k++)
			f[i][j][k]=inf;
			cnt[i][j]=j-i+1;
		}
	}
	for(int i=1;i<=n;i++)
	{
		int x;
		cin>>x;
		f[i][i][x]=0;
	}
    for(int len=2;len<=n;len++)
    {
    	for(int i=1;i+len-1<=n;i++)
    	{
    		int j=i+len-1;
            for(int k=0;k<3;k++)
            {
            	int minv=inf;
            	for(int t=i;t<j;t++)
            	{
            		if(f[i][t][k]==inf||f[t+1][j][k]==inf)continue;
            		minv=min(minv,f[i][t][k]+f[t+1][j][k]);
            	}
            	if(minv==inf)continue;
            	cnt[i][j]=1;
            	f[i][j][(k+1)%3]=min(f[i][j][(k+1)%3],minv+s[j]-s[i-1]);
            	w[i][j]=min({f[i][j][0],f[i][j][1],f[i][j][2]});
            }
    	      
    	}
    }
    for(int len=2;len<=n;len++)
    {
    	for(int i=1;i+len-1<=n;i++)
    	{
              int j=i+len-1;
              for(int t=i;t<j;t++)
              {
              	if(cnt[i][j]>cnt[i][t]+cnt[t+1][j])
              	{
              		cnt[i][j]=cnt[i][t]+cnt[t+1][j];
              		w[i][j]=w[i][t]+w[t+1][j];
              	}
              	if(cnt[i][j]==cnt[i][t]+cnt[t+1][j])
              	{
              		w[i][j]=min(w[i][j],w[i][t]+w[t+1][j]);
              	}
              }
    	      
    	}
    }
    cout<<cnt[1][n]<<' '<<w[1][n]<<endl;
}

signed main()
{
    ios::sync_with_stdio(false);
    int t=1;//cin>>t;
    while(t--)solve();
    return 0;
}