HDU4536 XCOM Enemy Unknown(dfs）

题目链接。

分析：

用dfs枚举每一波攻击的三个国家。

很暴力，但没想到0ms。

#include <iostream>
#include <cstdio>
#include <vector>

using namespace std;

const int maxn = ;

int p[maxn], nervous[maxn], n, m, k, max_cnt;
int att[][];

void dfs(int cn) {
    //界限
    if(max_cnt >= k) return ;
    if(cn >= k) {
        max_cnt = k; return ;
    }

    int a[], bak[];
    for(int v=; v<; v++) {    //枚举支援的国家
        /*
            * 被支援的国家为 att[cn][v]
            * 其他两个国家为 att[cn][(v+1)%3] 和 att[cn][(v+2)%3]
        */
        a[] = att[cn][v];  //将国家序号存起来，方便调用
        a[] = att[cn][(v+)%];
        a[] = att[cn][(v+)%];

        bak[] = nervous[a[]];  //备份
        bak[] = nervous[a[]];
        bak[] = nervous[a[]];

        nervous[a[]] -= ;
        if(nervous[a[]] < ) nervous[a[]] = ;
        nervous[a[]] += ;
        nervous[a[]] += ;

        //数据已经保证受攻击的3个国家一定在不同的洲
        for(int i=; i<n; i++) {
            if(p[i] == p[a[]] && i != a[]) nervous[i]++;
            if(p[i] == p[a[]] && i != a[]) nervous[i]++;
        }

        //价差是否都小于5
        int flag = true;
        for(int i=; i<n; i++) {
            if(nervous[i] > ) {
                max_cnt = max(max_cnt, cn);
                flag = false;
            }
        }

        if(flag) dfs(cn+);

        //还原
        for(int i=; i<n; i++) {
            if(p[i] == p[a[]] && i != a[]) nervous[i]--;
            if(p[i] == p[a[]] && i != a[]) nervous[i]--;
        }

        nervous[a[]] = bak[];
        nervous[a[]] = bak[];
        nervous[a[]] = bak[];
    }
}

int main(){
    int T;

    scanf("%d", &T);

    for(int kase=; kase<=T; kase++) {
        scanf("%d%d%d", &n, &m, &k);

        for(int i=; i<n; i++) {
            scanf("%d", &p[i]);
        }

        for(int i=; i<n; i++) {
            scanf("%d", &nervous[i]);
        }

        for(int i=; i<k; i++) {
            scanf("%d %d %d", &att[i][], &att[i][], &att[i][]);
        }

        max_cnt = ;
        printf("Case #%d: ", kase);

        //搜索
        dfs();

        printf("%d\n", max_cnt);
    }

    return ;
}