题意:给你一堆数,求问怎么分成两组,使两组的和相同,可以只选部分的数
思路:
这题首先num的范围特别大,50万,MLE了几次,最后发现要用滚动数组,而且在几次MLE的时候想到了,题目给了这些数的和是不超过500000的,所有当有一个数是大于250000的时候是没有用的,剩下的数的和不可能有这么大
dp[i][j] 表示前i个数第一组比另一组 高 j - 250000 的第一组的最大高度
那么状态转移方程就很好想了
从下向上更新就行了
AC代码如下:
#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
using namespace std;
int dp[2][550000];
int num[55], N, cnt;
int main(){
int T, Case = 1;
scanf( "%d", &T );
while( T-- ){
scanf( "%d", &N );
cnt = 0;
for( int i = 1; i <= N; i++ ){
int temp;
scanf( "%d", &temp );
if( temp <= 250000 ){
num[++cnt] = temp;
}
}
memset( dp, -1, sizeof( dp ) );
int now = 0;
int pre = 1;
dp[0][250000] = 0;
for( int i = 0; i < cnt; i++ ){
swap( now, pre );
for( int j = 0; j <= 500000; j++ ){
if( dp[pre][j] != -1 ){
if( j + num[i+1] <= 500000 ) dp[now][j+num[i+1]] = max( dp[now][j+num[i+1]], dp[pre][j] + num[i+1] );
if( j - num[i+1] >= 0 ) dp[now][j-num[i+1]] = max( dp[now][j-num[i+1]], dp[pre][j] );
dp[now][j] = max( dp[now][j], dp[pre][j] );
}
}
}
printf( "Case %d: ", Case++ );
if( dp[now][250000] <= 0 ){
cout << "impossible" << endl;
}else{
cout << dp[now][250000] << endl;
}
}
return 0;
}