HDU 5890 Eighty seven（dp+bitset）

最新推荐文章于 2020-12-17 01:19:05 发布

原创最新推荐文章于 2020-12-17 01:19:05 发布 · 251 阅读

1 ·

本内容遵循CC 4.0 BY-SA版权协议

HDU 同时被 3 个专栏收录

803 篇文章

订阅专栏

杂题

256 篇文章

订阅专栏

202 篇文章

订阅专栏

本文介绍了一种使用动态规划解决特定数值组合问题的方法。给定一系列整数，在删除三个数后，判断是否能从剩余的数中选出十个数，使它们的总和等于87。文章详细解释了通过枚举和状态转移方程来解决问题的过程，并提供了实现代码。

Description

给出 $n$ 个数 $a_1,...,a_n$ ，每次查询删去三个数（可能重复），问是否可能从剩下的数中取十个数使其和为 $87$

Input

第一行一整数 $T$ 表示用例组数，每组用例首先输入一整数 $n$ ，之后输入 $n$ 个整数 $a_1,...,a_n$ ，之后输入一整数 $q$ 表示查询数，每组查询输入三个整数 $i,j,k$ 表示删去 $a_i,a_j,a_k$ $(1\le T\le 5,1\le n\le 50,1\le q\le 10^5)$

Output

对于每组查询，如果删去 $a_i,a_j,a_k$ 后可以从剩余数中拿十个使得其和为 $87$ 则输出 $Yes$ ，否则输出 $No$

Sample Input

1
12
1 2 3 4 5 6 7 8 9 42 21 22
10
1 2 3
3 4 5
2 3 2
10 10 10
10 11 11
10 1 1
1 2 10
1 11 12
1 10 10
11 11 12

Sample Output

No
No
No
Yes
No
Yes
No
No
Yes
Yes

Solution

枚举删去的三个数 $a_i,a_j,a_k$ ，以 $dp[r][s][t]$ 表示从前 $r$ 个数中取 $s$ 个数和为 $t$ 的方案是否存在，如果 $r=i/j/k$ 则第 $r$ 个数不选，否则根据第 $r$ 个数是否选有转移 $dp[r][s][t]|=dp[r-1][s-1][t-a_r]|dp[r-1][s][t]$ ，时间复杂度 $O(880n^4)$ ，用 $bitset$ 优化一下第三维即可，预处理出所有情况之后对于查询 $O(1)$ 输出即可

Code

#include<cstdio>
#include<algorithm>
#include<bitset>
using namespace std;
bitset<88>dp[11];
int T,n,q,a[55],mark[55][55][55];
int check(int i,int j,int k)
{
    for(int i=0;i<=10;i++)dp[i].reset();
    dp[0][0]=1;
    for(int s=1;s<=n;s++)
        if(s!=i&&s!=j&&s!=k)
            for(int t=10;t>=1;t--)
                dp[t]|=dp[t-1]<<a[s];
    if(dp[10][87])return 1;
    return 0;
}
int main()
{
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d",&n);
        for(int i=1;i<=n;i++)scanf("%d",&a[i]);
        for(int i=1;i<=n;i++)
            for(int j=i;j<=n;j++)
                for(int k=j;k<=n;k++)
                    mark[i][j][k]=check(i,j,k);
        scanf("%d",&q);
        while(q--)
        {
            int b[3];
            scanf("%d%d%d",&b[0],&b[1],&b[2]);
            sort(b,b+3);
            printf("%s\n",mark[b[0]][b[1]][b[2]]?"Yes":"No");
        }
    }
    return 0;
}