HDU 5616 Jam's balance（dp）

Description
Jam有N个砝码和一个没有游标的天平，现在给他(1≤N≤20)个砝码，砝码可以放左边，也可以放右边，问可不可以测出所问的重量, 问的个数为(1≤M≤100)个.
Input
第一行T(1≤T≤5)，表示T组数据。
接下来T组数据：
接下来第一行一个数N，表示砝码个数。
接下来第二行N个数，表示砝码们的重量(1≤w​i≤100)。
接下来第三行一个数M，表示询问个数。
接下来M行每行一个数k，表示一个询问。
Output
对于每组数据，输出”YES”或者”NO”
Sample Input
1
2
1 4
3
2
4
5
Sample Output
NO
YES
YES
Solution
mark[i][j]表示前i个砝码是否能够称出重量j，则有转移方程
mark[i-1][j]=1 -> mark[i][j+a[i]]=mark[i][j-a[i]]=1
Code

#include<cstdio>
#include<iostream>
#include<cstring>
using namespace std;
typedef long long ll;
#define INF 0x3f3f3f3f
#define maxn 2222
int T,n,a[maxn],flag[maxn],mark[maxn],q,m;
int main()
{
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d",&n);
        memset(flag,0,sizeof(flag));
        memset(mark,0,sizeof(mark));
        for(int i=1;i<=n;i++)scanf("%d",&a[i]);
        flag[0]=1;
        for(int i=1;i<=n;i++)
        {
            for(int j=0;j<=2000;j++)
                if(flag[j])
                {
                    if(j-a[i]>=0)mark[j-a[i]]=1;
                    if(j+a[i]<=2000)mark[j+a[i]]=1;
                }
            for(int j=0;j<=2000;j++)flag[j]|=mark[j];
        }
        scanf("%d",&q);
        while(q--)
        {
            scanf("%d",&m);
            if(m>2000||!flag[m])printf("NO\n");
            else printf("YES\n");
        }
    }
    return 0;
}