UVALive 6669 Hidden Tree DP

很有意思的DP！

题目大意：给一个序列a[]，你从中选择一些子序列，将子序列的值从左往右依次放到某棵二叉树的叶子节点上，使得除了叶子，所有节点左右子树权和相等。子树的权和 = 子树叶子的权和。如果存在这样一棵二叉树，选择的子序列就是合法的。问，最长的合法子序列是多少。

思路：枚举二叉树可能的叶子的最小权，显然，能和此数一起组成二叉树的数，要么和这个数相等，要么是这个数的2^k倍。把满足这种关系的数，认做一个集合，显然集合外的数，不能和集合内的数组成二叉树。那么，我们只需要一个一个得求出所有集合的最长子序列即可。

把集合内的所有数全部除以最小权，剩下的数为1,2,4,8,16,32,64.....这种2^k的数。假设你从左到右，第一个填的数为16，第二个填的数一定不会比16大，不然那个16无法合并。如果填的就是16，那就合成为32。当然，填小于16的数也是行的。那么，对于2^k的数，每个数，一定只有两种状态，有1个，或者没有。那么我们似乎就可以用状态压缩就可，一个难点就是状态的上限值的设定，如果只考虑最大值的话，感觉太大，但是其实考虑和值，其实不是很大。至此，此题就可以做了。

//#pragma comment(linker, "/STACK:102400000,102400000")
#include<cstdio>
#include<cstring>
#include<vector>
#include<queue>
#include<cmath>
#include<cctype>
#include<string>
#include<algorithm>
#include<iostream>
#include<ctime>
#include<map>
#include<set>
using namespace std;
#define MP(x,y) make_pair((x),(y))
#define PB(x) push_back(x)
typedef long long LL;
//typedef unsigned __int64 ULL;
/* ****************** */
const int INF=100011122;
const double INFF=1e100;
const double eps=1e-8;
const LL mod=1000000007;
const int NN=1005;
const int MM=1000010;
/* ****************** */

int dp[NN*505];
int a[NN],b[NN],sum[NN];
bool yes[NN],pp2[NN*505];

int solve(int x,int n)
{
    int i,j,t,tol=0,ans=0;
    for(i=1;i<=n;i++)
    {
        if(a[i]%x==0 && pp2[a[i]/x])
            b[++tol]=a[i]/x;
    }
    sum[0]=0;
    for(i=1;i<=tol;i++)
        sum[i]=sum[i-1]+b[i];

   // printf("tol==%d\n",tol);
   // for(i=1;i<=tol;i++)
    //    printf("%d%c",b[i],i==tol?'\n':' ');

    for(j=0;j<=sum[tol];j++)dp[j]=-111122200;
    dp[0]=0;
    for(i=1;i<=tol;i++)
        for(j=sum[i];j>=b[i];j--)
        {
            t=j-b[i];
            if( (t&(b[i]-1))==0)
            {
                dp[j]=max(dp[j],dp[t]+1);
            }
        }

    for(j=1;j<=sum[tol];j++)
        if(pp2[j])
        {
            ans=max(ans,dp[j]);
        }
    return ans;
}

int main()
{
    int i,j,n,t,ans;
    memset(pp2,false,sizeof(pp2));
    pp2[i=1]=true;

    while(i*2<NN*505)
    {
        i=i*2;
        pp2[i]=true;
    }
    while(scanf("%d",&n)!=EOF)
    {
        if(n==0)break;

        ans=1;
        for(i=1;i<=n;i++)
            scanf("%d",&a[i]);
        for(i=1;i<=n;i++)
        {
            yes[i]=true;
            for(j=1;j<=n;j++)
            {
                if(a[i]!=a[j] && a[i]%a[j]==0 && pp2[a[i]/a[j]])
                {
                    yes[i]=false;
                    break;
                }
            }
        }
        for(i=1;i<=n;i++)
        {
            if(yes[i])
            {
                t=solve(a[i],n);
                ans=max(ans,t);
                for(j=1;j<=n;j++)
                    if(a[j]==a[i])
                        yes[j]=false;
            }
        }
        printf("%d\n",ans);
    }
    return 0;
}