UVa 1289 Stacking Plates
题目
题目大意
有 N N 堆盘子,第堆盘子有 hi h i 个,从上到下直径不减。有两种操作:
- split:将一堆盘子从某个位置分成上下两堆;
- join:将一堆盘子 a a 放在另一堆盘子上,要求 a a 底部的盘子直径不超过顶部盘子直径。
求将所有盘子叠成一堆的最少操作步数。
思路
仔细分析题目就会发现:最少操作步数=拆分次数+合并次数-N-1=拆分次数*2-N-1。
则该问题转化为求最少的拆分次数。
将盘子排序,大小相同的盘子视为同一盘子。
定义状态 f[i][j] f [ i ] [ j ] 为到第 i i 种盘子,这个盘子来自第堆的最少拆分次数。
记 c(i) c ( i ) 为第 i i 种盘子在所有盘子中的个数
则可枚举上一个盘子所在的堆。若 k k 可以放在第种盘子的底部和第 i i 种盘子的顶部,则可转移至 f[i−1][k]+c(i)−1 f [ i − 1 ] [ k ] + c ( i ) − 1 ,否则转移至 f[i−1][k]+c(i) f [ i − 1 ] [ k ] + c ( i ) 。
注意 k k 需要满足的条件:第堆不仅需要拥有第 i−1 i − 1 种盘子,且需要拥有第 i i 种盘子。除非第种盘子的来源只有 j j ,否则必须满足。
答案即为 f[M][i](1≤i≤N) f [ M ] [ i ] ( 1 ≤ i ≤ N ) ,其中 M M <script type="math/tex" id="MathJax-Element-2729">M</script>为不同种类的盘子数量。
正解代码
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int Maxn=50;
const int Maxh=50;
const int INF=0x3f3f3f3f;
int N,c,M;
int cnt[Maxn*Maxh+5];//记录第i种盘子出现次数
int f[Maxn*Maxh+5][Maxn+5];
int h[Maxn*Maxh+5][Maxn+5];//记录第i种盘子是否在第j堆出现
pair<int,int> A[Maxn*Maxh+5];//所有盘子,first记录直径,second记录来自第几堆
void Prepare() {
memset(cnt,0,sizeof cnt);
memset(h,0,sizeof h);
memset(f,0x3f,sizeof f);
sort(A+1,A+c+1);
c=unique(A+1,A+c+1)-A-1;//去掉相同盘子
for(int i=1;i<=c;i++) {
int p=i;M++;
while(A[p].first==A[p+1].first&&p<c)
p++;
for(int j=i;j<=p;j++)
h[M][A[j].second]=1;
cnt[M]=p-i+1;i=p;
}//处理出h数组及cnt数组
}
int Solve() {
for(int i=1;i<=N;i++)
if(h[1][i])f[1][i]=cnt[1];
//初始化,若第i堆有第1种盘子,则该状态赋值为cnt[1],否则赋值为INF
for(int i=2;i<=M;i++)
for(int j=1;j<=N;j++)
if(h[i][j])//注意有这种盘子时才能尝试转移
for(int k=1;k<=N;k++)
if(h[i][k]&&(cnt[i]==1||k!=j))
f[i][j]=min(f[i][j],f[i-1][k]+cnt[i]-1);
else f[i][j]=min(f[i][j],f[i-1][k]+cnt[i]);
int ret=INF;
for(int i=1;i<=N;i++)
ret=min(ret,f[M][i]);
//找出最小拆分次数
return ret*2-N-1;//返回答案
}
int main() {
#ifdef LOACL
freopen("in.txt","r",stdin);
freopen("out.txt","w",stdout);
#endif
int cas=0;
while(scanf("%d",&N)!=EOF) {
c=M=0;
for(int i=1;i<=N;i++) {
int x;
scanf("%d",&x);
for(int j=1;j<=x;j++) {
scanf("%d",&A[++c].first);
A[c].second=i;
}
}
Prepare();
printf("Case %d: %d\n",++cas,Solve());
}
return 0;
}