描述
2001年9月11日,一場突發的災難將紐約世界貿易中心大廈夷爲平地,Mr. F曾親眼目睹了這次災難。爲了紀念“9?11”事件,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”。
輸入:
5 1 3 4 5 2
輸出:
7
思路: 揹包
dp[i][j] 表示的是 前i個水晶, 高度差爲j, 高塔的高度。
對於第i個水晶, 有3種情況。
1、不放
2、放高塔
3、放矮塔
(1).高塔仍高
(2).矮塔變高塔
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <iostream>
using namespace std;
const int v = 2000 + 5;
const int MaxN = 100 + 5;
int N, sum, num[MaxN], dp[2][v];
int main() {
int i, j, k;
cin >> N;
for(i = 0; i < N; ++i) {
cin >> num[i];
sum += num[i];
}
memset(dp, -1, sizeof(dp));
dp[1][0] = 0;
int a;
for(i = 0; i < N; ++i) {
a = i % 2;
memset(dp[a], -1, sizeof(dp[a]));
for(j = 0; j <= sum; ++j) {
//1.不放第i塊水晶;
if(dp[a ^ 1][j] > -1)
dp[a][j] = dp[a ^ 1][j];
//2.放進去後,高塔變矮塔(第i塊放在矮塔上了);
if(num[i] > j && dp[a ^ 1][num[i] - j] > -1)
dp[a][j] = max(dp[a][j], dp[a ^ 1][num[i] - j] + j);
//3.放進去後,高塔仍高(第i塊放在矮塔上);
if(j + num[i] <= sum && dp[a ^ 1][j + num[i]] > -1)
dp[a][j] = max(dp[a][j], dp[a ^ 1][j + num[i]]);
//4.放進去後,高塔更高(第i塊放在高塔上).
if(j >= num[i] && dp[a ^ 1][j - num[i]] > -1)
dp[a][j] = max(dp[a][j], dp[a ^ 1][j - num[i]] + num[i]);
}
}
if(dp[a][0] > 0)
cout << dp[a][0] << endl;
else
cout << "Impossible" << endl;
}