题意:给一个序列a[],你从中选择一些子序列,将子序列的值从左往右依次放到某棵二叉树的叶子节点上,使得除了叶子,所有节点左右子树权和相等。子树的权和 = 子树叶子的权和。如果存在这样一棵二叉树,选择的子序列就是合法的。问,最长的合法子序列是多少。
思路:考虑最后的合法子序列,其中的每一个元素肯定是base*2^k的形式,所以对于每一个奇数base我们分别求解然后求出最大值即可,那么也就是说现在问题转化为了“一个由2的幂组成的序列选出一个子序列使其合法”。
假设当前已选择了元素k作为当前序列的最后一个元素,那么之后的下一个元素一定不能大于k,这是因为如果下一个元素大于k,那么k这个元素将无法和其他元素结合,那么就无法组成一棵合法的二叉树,换句话说,对于当前的元素,只能选择小于等于他的元素,所以对于当前元素k,在他之前比他大的元素最多只有1个,可以用一位二进制表示这个状态。
因为每个数小于等于500,所以在转化后的序列中每个元素最大为256,共有1000个元素,所以说最大可能达到的合法状态是256*512,虽然这样状态数可能很多,但是由于每次转移我们只转移部分状态,所以时间复杂度并不大。
状态转移方程为
dp[i+a[i]] = max(dp[i+a[i]], dp[i]+1), d[i] > 0
dp[a[i]] = max(dp[a[i]], 1), a[i]为序列中的元素(2的幂的形式)
注意要和背包问题一样从大到小递推,这样才能保证当前的状态是由上一步的状态递推而来。
#include<cstdio>
#include<cstring>
#include<cmath>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#include<vector>
#include<map>
#include<queue>
#include<stack>
#include<string>
#include<map>
#include<set>
#include<ctime>
#define eps 1e-6
#define LL long long
#define pii pair<int, int>
//#pragma comment(linker, "/STACK:1024000000,1024000000")
using namespace std;
const int MAXN = 1010;
int n;
int b[MAXN];
int dp[1000000];
int DP(int num, int a[]) {
int sumv = 0, mi = 0;
for(int i = 1; i <= num; i++) sumv += a[i];
while((1<<(mi+1)) <= sumv) mi++;
int maxs = 1 << mi;
dp[0] = 1;
for(int i = 1; i <= num; i++) {
int tmp = a[i];
for(int i = maxs; i >= tmp; i -= tmp) {
if(dp[i]) dp[i+tmp] = max(dp[i+tmp], dp[i]+1);
}
dp[tmp] = max(dp[tmp], 1);
}
int ans = 0;
for(int i = 0; i <= mi; i++) ans = max(ans, dp[1<<i]);
for(int i = 1; i <= num; i++) {
int tmp = a[i];
for(int i = maxs; i >= tmp; i -= tmp) {
dp[i+tmp] = 0;
}
dp[tmp] = 0;
}
return ans;
}
int a[MAXN];
int main() {
//freopen("input.txt", "r", stdin);
while(scanf("%d", &n)==1 && n) {
int maxb = -1;
for(int i = 1; i <= n; i++) scanf("%d", &b[i]), maxb = max(maxb, b[i]);
int ret = 0;
for(int i = 1; i <= maxb; i+=2) {
int sz = 0;
for(int j = 1; j <= n; j++) {
if(b[j]%i==0) {
int t = b[j] / i;
for(int k = 0; k < 11; k++) {
if((1<<k) == t) {
a[++sz] = t;
break;
}
}
}
}
if(sz) ret = max(ret, DP(sz, a));
}
cout << ret << endl;
}
return 0;
}