【NOIP】石子合并
题目描述
在一个圆形操场的四周摆放 N 堆石子,现要将石子有次序地合并成一堆.规定每次只能选相邻的2堆合并成新的一堆,并将新的一堆的石子数,记为该次合并的得分。
试设计出一个算法,计算出将 N 堆石子合并成 1 堆的最小得分和最大得分。
输入格式
数据的第 1 行是正整数 N表示有N 堆石子。
第 2 行有 N 个整数,第 i 个整数 a_i表示第 i 堆石子的个数。
输出格式
输出共 2行,第 1 行为最小得分,第 2 行为最大得分。
输入输出样例
输入 #1 复制
4
4 5 9 4
输出 #1 复制
43
54
代码
这类dp题, 题目说的是:
可以从左到右顺序每次合并前两堆, 也可以 左边的合成一堆,右边的合成一堆,最后合并这两个大堆。
我们需要用到的,就是后面这种,这种用来求 最大,最小 都可以。
1:这种dp,称为,区间dp,一个大的分成左右两个区间,左右区间里 又分 左右区间。。。。。。。
小区间问题解决了,大的也就能解决了。
2:先做预处理,把各种堆的数据先算好 cost【l】【r】(l—r 堆这间的合)
转移方程:求最小: dp1[l][r] = min(dp1[l][r],dp1[l][k]+dp1[k+1][r]+cost[l][r]); (左区间 + 右区间 + 总的合)
求最大也是一样。
注意:
这是题目说的是 围成一圈,所以是环形的, 我们需要开两倍n,相当于把n个 复制一遍,排成1–n ,1—n模拟环形
ps:如果不太懂,可以先把问题简单化,比如:取消循环条件, 而且只求 最小得分。
代码 下面写了这个简单的问题。如果不懂可以先弄懂简单的。
#include<iostream>
using namespace std;
#include<cstring>
int n; //因为是循环,所以数组都要开两倍
int w[205]; //每堆石子的原重量。
int dp1[205][205]; //做最小得分
int dp2[205][205]; //做最大得分
int cost[205][205]; //cost[i][j] 表示 第i--j堆 的石子的总重量
int main(){
cin>>n;
for(int i=1;i<=n;i++){
cin>>w[i];
w[i+n] = w[i]; //复制成2*n,模拟循环
}
// memset(dp1,1e6,sizeof(dp1)); //因为要求最小,先给初值,
// 求最大的就不需要额,
for(int i=1;i<=n*2;i++){ //这次要开两倍 ,n*2
for(int j=i;j<=n*2;j++){
for(int k=i;k<=j;k++){
cost[i][j] += w[k];
}
}
}
for(int i=1;i<=n;i++){
dp1[i][i] = 0;
dp2[i][i] = 0;
}
for(int len=1;len<n;len++){
for(int l=1,r=l+len;r<n*2&&l<n*2;l++,r=l+len){
dp1[l][r] = 1e8; //也可以在这里随便给个很大的值就可以了
for(int k=l;k<r;k++){
dp1[l][r] = min(dp1[l][r],dp1[l][k]+dp1[k+1][r]+cost[l][r]);
dp2[l][r] = max(dp2[l][r],dp2[l][k]+dp2[k+1][r]+cost[l][r]);
}
}
}
int ans1 = 1e8,ans2=0;
for(int i=1;i<=n;i++){
ans1 = min(ans1,dp1[i][n+i-1]);
ans2 = max(ans2,dp2[i][n+i-1]);
}
cout<<ans1<<endl<<ans2;
return 0;
}
//问题简化下。取消循环条件, 求最小得分
//#include<iostream>
//using namespace std;
//#include<cstring>
//int n;
//int w[105]; //每堆石子的原重量。
//int dp[105][105]; //做dp
//int cost[105][105]; //cost[i][j] 表示 第i--j堆 的石子的总重量
//int main(){
// cin>>n;
// for(int i=1;i<=n;i++){
// cin>>w[i];
// }
// memset(dp,1e6,sizeof(dp));
// for(int i=1;i<=n;i++){
// for(int j=i;j<=n;j++){
// for(int k=i;k<=j;k++){
// cost[i][j] += w[k];
// }
// }
// }
// for(int i=1;i<=n;i++){
// dp[i][i] = 0;
// }
// for(int len=2;len<=n;len++){
// for(int l=1;l<=n-len+1;l++){
// int r = l+len-1;
// for(int k=l;k<=r;k++){
// dp[l][r] = min(dp[l][r],dp[l][k]+dp[k+1][r]+cost[l][r]);
// }
// }
// }
// cout<<dp[1][n];
// return 0;
//}