UVa 10400 记忆化搜索

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;

const int INF = ;
const int maxn = ;

int a[maxn];
char ans[maxn]; //a[i] = 1代表+，2代表-，3代表*，4代表/  ;
int len,goal;
bool vis[maxn][INF+][];  //  vis[i][j][1]=1 表示计算完前i个数字的某种组合可以得到正j，dp[i][j][0]=1则为负j

bool dfs(int u,int sum){
    if(u == ){
        if(a[] == sum ) return true;
        else             return false;
    }

    if(sum >=  && vis[u][sum][])
      return false;

    else if(sum< && vis[u][-sum][])
      return false;

    int temp = sum - a[u];
    if(temp > -INF && temp < INF){
        ans[u] = '+';
        if(dfs(u-,temp))   return true;
        if(temp >= ){
            vis[u-][temp][] = true;
        }
        else{
            vis[u-][-temp][] = true;
        }
    }
    temp = sum + a[u];
    if(temp > -INF && temp < INF){
        ans[u] = '-';
        if(dfs(u-,temp))   return true;
        if(temp >= ){
            vis[u-][temp][]  = true;
        }
        else{
            vis[u-][-temp][] = true;
        }
    }
    temp = sum * a[u];
    if(temp > -INF && temp < INF){
        ans[u] = '/';
        if(dfs(u-,temp))   return true;
        if(temp >= ){
            vis[u-][temp][]  = true;
        }
        else{
            vis[u-][-temp][] = true;
        }
    }
    if(sum % a[u]) return false;
    temp = sum / a[u];
    if(temp > -INF && temp < INF){
        ans[u] = '*';
        if(dfs(u-,temp))   return true;
        if(temp >= ){
            vis[u-][temp][]  = true;
        }
        else{
            vis[u-][-temp][] = true;
        }
    }
    return  false;
}
void print(){
    ans[len+] = '=';
    for(int i=;i<=len;i++){
        printf("%d%c",a[i],ans[i+]);
    }
    printf("%d\n",goal);
}
int main()
{
   //freopen("E:\\acm\\input.txt","r",stdin);
    int T;
    cin>>T;
    while(T--){
        scanf("%d",&len);
        for(int i=;i<=len;i++)
            scanf("%d",&a[i]);
        scanf("%d",&goal);

        memset(vis,,sizeof(vis));

        if(dfs(len,goal)){
            print();
        }
        else
            printf("NO EXPRESSION\n");
    }
}