AtCoder Beginner Contest 310 D

题目链接：D - Peaceful Teams (atcoder.jp)

从数据范围来看 考虑用暴搜

每次加入一个人     时间复杂度O（N * N！）

#include <bits/stdc++.h>
using namespace std;

#define ll long long
#define qwq ios::sync_with_stdio(0), cin.tie(0), cout.tie(0)
const int N = 20, M = 50;
vector <int> a(M), b(M); // a[i],b[i]不能在同一组
int ans, n, T, m;
vector <int> c(N); // 记录第 i 个人所在的组
void dfs(int p, int t) { //  p 是当前安排第p个人
    //  t 是当前可以安排的组 确保 1 ~ t - 1组都已经有人 第t组可不一定有人
    if (p == n + 1) {
        if (t != T + 1) return; // 没有安排满 T 组 直接返回
        for (int i = 1; i <= m; ++ i) 
            if (c[a[i]] == c[b[i]]) return; // 将矛盾的人安排在一组了 返回
        ++ ans; // 条件判断都满足，则方案数 + 1
        return;
    }
    for (int i = 1; i <= t; ++ i) {
        c[p] = i;
        dfs(p + 1, max(t, i + 1)); // 如果可以创建新的组（即i遍历到t） 则创建新组
    }
}
void solve() {
	
    cin >> n >> T >> m;
    for (int i = 1; i <= m; ++ i) {
        cin >> a[i] >> b[i];
    // 如果dfs 从 0 开始， 此处需要a[i]--, b[i]--
    }
    //从第一个人开始安排 此时能安排第一组
    dfs(1, 1);
    cout << ans;
}
int main(){
	qwq; int T = 1;
	//cin >> T;
	while (T -- ) 
		solve();
	
	return 0;
}

 使用vector ：

#include <bits/stdc++.h>
using namespace std;

#define ll long long
#define qwq ios::sync_with_stdio(0), cin.tie(0), cout.tie(0)
const int N = 20, M = 50;
int g[N][N];
int ans, n, T, m;

vector <vector <int> > v;

void dfs(int i) {
    if (i > n) {
        if (v.size() == T) ++ ans;
        return;
    }
    for (int t = 0; t < v.size(); ++t) {
        bool ok = 1; // 判断能否放进第 t 组 (该种做法是0 ~ T - 1组)
        for (auto j : v[t]) 
            if (g[i][j]) ok = 0; // 如果有矛盾， 则不能放进第t 组
        if (ok) {
            v[t].push_back(i);
            dfs(i + 1);
            v[t].pop_back();
        }
    }
    v.push_back(vector<int> (1, i)); // 创建一个新的组
    // vector<int> (1, i)表示创建了一个 vector<int> 类型的对象，并初始化为包含一个元素 i 的向量。使得v.size() + 1
    dfs(i + 1);
    v.pop_back(); // 回溯
}
void solve() {
	v.clear();
    cin >> n >> T >> m;
    for (int i = 1, a, b; i <= m; ++ i) {
        cin >> a >> b;
        g[a][b] = g[b][a] = 1;
    }
    //从第一个人开始安排
    dfs(1);
    cout << ans;
}
int main(){
	qwq; int T = 1;
	//cin >> T;
	while (T -- ) 
		solve();
	
	return 0;
}