目录

牛客_体操队形_dfs+枚举

题目解析

C++代码

Java代码

---

牛客_体操队形_dfs+枚举

体操队形 (nowcoder.com)

描述：
        dd作为体操队队长，在给队员们排队形，体操队形为一个单独的纵列，体操队有n个同学，标号为1∼n，对于i(1≤i≤n)号队员，会有一个诉求(1≤a[i]≤n)，表示他想排在a[i]号队员前面，当a[i]=i时，我们认为他没有位置需求，随便排哪儿都行，dd想知道有多少种队形方案，可以满足所有队员的要求。

输入描述：

读入第一行一个数字n(2≤n≤10)
第二行n个数字，表示a[i],保证1≤a[i]≤n

输出描述：

输出一行，表示方案数

---

题目解析

        根据题目中的队员位置需求与位置是否被占进行递归减治策略，采用dfs依次搜索合法位置就入队，知道所有队员全部放入即可。接下来就是程序实现。

C++代码

#include <iostream>
#include <vector>
using namespace std;int n = 0, res = 0;void dfs(vector<int>& arr, int pos, vector<bool>& vis)
{if(pos == n + 1) // 找到一个合法方案{++res;return;}for(int i = 1; i <= n; ++i){if(vis[i] == true) // 当前i已经放过了，剪枝continue;if(vis[arr[i]] == true) // 当前i的诉求已经放过了，剪枝return;vis[i] = true; // 放上idfs(arr, pos + 1, vis);vis[i] = false; // 回溯}
}int main()
{cin >> n;vector<int> arr(n + 1);vector<bool> vis(n + 1);for(int i = 1; i <= n; ++i){cin >> arr[i];}dfs(arr, 1, vis);cout << res << endl;return 0;
}

Java代码

import java.util.*;
public class Main
{public static int N = 15;public static int n, ret;public static boolean[] vis = new boolean[N];public static int[] arr = new int[N];public static void dfs(int pos){if(pos == n + 1) // 找到⼀种合法情况{ret++;return;}for(int i = 1; i <= n; i++){if(vis[i] == true) continue; // 剪枝 - i 号队员已经放过了if(vis[arr[i]]) return; // 剪枝vis[i] = true; // 相当于已经放上 i 号队员dfs(pos + 1);vis[i] = false; // 回溯 - 恢复现场}}public static void main(String[] args){Scanner in = new Scanner(System.in);n = in.nextInt();for(int i = 1; i <= n; i++){arr[i] = in.nextInt();}dfs(1);System.out.println(ret);}
}