洛谷 P2423 双塔
题目
题目描述
2001年9月11日,一场突发的灾难将纽约世界贸易中心大厦夷为平地,Mr. F曾亲眼目睹了这次灾难。为了纪念“911”事件,Mr. F决定自己用水晶来搭建一座双塔。Mr. F有N块水晶,每块水晶有一个高度,他想用这N块水晶搭建两座有同样高度的塔,使他们成为一座双塔,Mr. F可以从这N块水晶中任取M(1≤M≤N)块来搭建。但是他不知道能否使两座塔有同样的高度,也不知道如果能搭建成一座双塔,这座双塔的最大高度是多少。所以他来请你帮忙。
给定水晶的数量N(1≤N≤100)和每块水晶的高度Hi(N块水晶高度的总和不超过2000),你的任务是判断Mr. F能否用这些水晶搭建成一座双塔(两座塔有同样的高度),如果能,则输出所能搭建的双塔的最大高度,否则输出“Impossible”。
输入输出格式
输入格式:
输入的第一行为一个数N,表示水晶的数量。第二行为N个数,第i个数表示第i个水晶的高度。
输出格式:
输出仅包含一行,如果能搭成一座双塔,则输出双塔的最大高度,否则输出一个字符串“Impossible”。
输入输出样例
输入样例#1:
5
1 3 4 5 2
输出样例#1:
7
题解
f[i][j]表示匹配到第i个水晶,两塔的高度差为j时,高的水晶塔的高度,最后输出答案即为f[n][0]
代码
#include<cstdio>
#include<cstring>
using namespace std;
int tot,n;
int a[105],f[2005][2005];
int readln()
{
int x=0;
char ch=getchar();
while (ch<'0'||ch>'9') ch=getchar();
while ('0'<=ch&&ch<='9') x=x*10+ch-48,ch=getchar();
return x;
}
int max(int x,int y)
{
if (x>y) return x; else return y;
}
int main()
{
n=readln();
for (int i=1;i<=n;i++) {a[i]=readln();tot+=a[i];}
memset(f,-0x3f3f3f3f,sizeof(f));
f[0][0]=0;
for(int i=1;i<=n;i++)
{
for(int j=tot;j>=0;j--)
{
f[i][j]=max(f[i][j],f[i-1][j]);
f[i][j]=max(f[i][j],f[i-1][j+a[i]]);
if (j>=a[i]) f[i][j]=max(f[i][j],f[i-1][j-a[i]]+a[i]);
if (j<=a[i]) f[i][j]=max(f[i][j],f[i-1][a[i]-j]+j);
}
}
if (f[n][0]==0) printf("Impossible\n"); else printf("%d\n",f[n][0]);
return 0;
}