uva 10604 化学药品链式dp

题意：

给一堆化学药品，可以相互反应，反应会产生热量，或消耗热量。

给每种化学药品的个数，求最终能达到的最低温度。

解析：

数量较少，直接dfs。

代码：

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <algorithm>
#include <cstring>
#include <cmath>
#include <stack>
#include <vector>
#include <queue>
#include <map>
#include <climits>
#include <cassert>
#define LL long long

using namespace std;
const int maxn = 1e6;
const int inf = 0x3f3f3f3f;
const double eps = 1e-8;
const double pi = 4 * atan(1.0);
const double ee = exp(1.0);

int n;
int heat[10][10];
int result[10][10];
int dp[10][10][10][10][10][10];
int vis[10][10][10][10][10][10];
int cnt[10];

int dfs(int a, int b, int c, int d, int e, int f)
{
    int& res = dp[a][b][c][d][e][f];
    if (vis[a][b][c][d][e][f])
        return res;
    vis[a][b][c][d][e][f] = true;

    res = inf;
    for (int i = 1; i <= 6; i++)
    {
        for (int j = 1; j <= 6; j++)
        {
            if ((i == j && 2 <= cnt[i]) || (i != j && 1 <= cnt[i] && 1 <= cnt[j]))
            {
                cnt[i]--;
                cnt[j]--;
                cnt[result[i][j]]++;
//                int t = dfs(cnt[1], cnt[2], cnt[3], cnt[4], cnt[5], cnt[6]);
//                if (t != inf)
//                    res = min(t + heat[i][j], res);
                res = min(res, dfs(cnt[1], cnt[2], cnt[3], cnt[4], cnt[5], cnt[6]) + heat[i][j]);
                cnt[i]++;
                cnt[j]++;
                cnt[result[i][j]]--;
            }
        }
    }
    if (res == inf)
        return res = 0;
    return res;
}

int main()
{
#ifdef LOCAL
    freopen("in.txt", "r", stdin);
#endif // LOCAL
    int ncase;
    scanf("%d", &ncase);
    while (ncase--)
    {
        scanf("%d", &n);
        memset(dp, 0, sizeof(dp));
        memset(vis, false, sizeof(vis));
        memset(cnt, 0, sizeof(cnt));
        memset(heat, 0, sizeof(heat));
        memset(result, 0, sizeof(result));
        for (int i = 1; i <= n; i++)
        {
            for (int j = 1; j <= n; j++)
            {
                int index, t;
                scanf("%d%d", &index, &t);
                result[i][j] = index;
                heat[i][j] = t;
            }
        }
        int k;
        scanf("%d", &k);
        for (int i = 1; i <= k; i++)
        {
            int x;
            scanf("%d", &x);
            cnt[x]++;
        }
        int ans = dfs(cnt[1], cnt[2], cnt[3], cnt[4], cnt[5], cnt[6]);
        printf("%d\n", ans);
        //getchar();
        //getchar();
        char s[10];
        scanf("%s", s);
    }
    return 0;
}