算法問題實戰策略 PICNIC

下面是另外一道搜索題目的解答過程
題目是《算法問題實戰策略》中的一題
oj地址是韓國網站 鏈接比較慢 https://algospot.com/judge/problem/read/PICNIC
大意以下

輸入
3 
2 1 
0 1 
4 6 
0 1 1 2 2 3 3 0 0 2 1 3 
6 10 
0 1 0 2 1 2 1 3 1 4 2 3 2 4 3 4 3 5 4 5

輸出
1
3
4

 

也是上來就擼一把DFS
所有可以匹配完成則計數增長1
可是有不少重複計算
我試過記錄關係對的時候 以數值大小爲序 只能排除一部分重複計算
錯誤的代碼:

 1 #include <iostream>
 2 #include <vector>
 3 #include <algorithm>
 4 
 5 
 6 using namespace std;
 7 
 8 /*
 9 3
10 2 1
11 0 1
12 4 6
13 0 1 1 2 2 3 3 0 0 2 1 3
14 6 10
15 0 1 0 2 1 2 1 3 1 4 2 3 2 4 3 4 3 5 4 5
16 //====================================
17 1
18 3
19 4
20 */
21 
22 int t;
23 int a, b;
24 int n, m;
25 typedef pair<int, int> PII;
26 
27 
28 bool isFriend(int i, int j, const vector<PII>& friendv)
29 {
30     int minIdx = min(i, j);
31     int maxIdx = max(i, j);
32 
33     for (auto& e : friendv) {
34         if (e.first == minIdx && e.second == maxIdx)
35             return true;
36     }
37 
38     return false;
39 }
40 
41 
42 int dfs(bool isChecked[],const vector<PII>& friendv)
43 {
44     int firstFree = -1;
45     for (int i = 0; i < n; i++) {
46         if (isChecked[i] == false){
47             firstFree = i;
48             break;
49         }
50     }
51 
52     if (-1 == firstFree)
53         return 1;
54 
55     int ret = 0;
56 
57 
58     for (int secondFree = firstFree + 1; secondFree < n; secondFree++) {
59         if (firstFree != secondFree && isChecked[firstFree] == false && isChecked[secondFree] == false 
60                 && isFriend(firstFree, secondFree, friendv)) {
61             isChecked[firstFree] = true; isChecked[secondFree] = true;
62             ret += dfs(isChecked, friendv);
63             isChecked[firstFree] = false; isChecked[secondFree] = false;
64         }
65     }
66 
67 
68     return ret;
69 }
70 
71 
72 
73 
74 int main()
75 {
76     cin >> t;
77 
78     while (t--) {
79         cin >> n >> m;
80         vector<PII> friendv;
81         bool isChecked[160] = {false};
82         while (m--) {
83             cin >> a >> b;
84             friendv.push_back({min(a,b),max(a,b)});
85         }
86         sort(friendv.begin(), friendv.end());
87         cout << dfs(isChecked, friendv);
88     }
89 
90 
91     return 0;
92 }

View Code

通過調試 發現DFS中雙重循環有很大問題
i=0時候檢測出 0 1配對 而後檢測出2 3 配對.
可是當i=2時 也能檢測2 3 配對 以及 0 1 配對。

因而作了如下修改，解決重複問題
ac代碼

 1 #include <iostream>
 2 #include <vector>
 3 #include <algorithm>
 4 
 5 
 6 using namespace std;
 7 
 8 /*
 9 3
10 2 1
11 0 1
12 4 6
13 0 1 1 2 2 3 3 0 0 2 1 3
14 6 10
15 0 1 0 2 1 2 1 3 1 4 2 3 2 4 3 4 3 5 4 5
16 //====================================
17 1
18 3
19 4
20 */
21 
22 int t;
23 int a, b;
24 int n, m;
25 typedef pair<int, int> PII;
26 
27 
28 bool isFriend(int i, int j, const vector<PII>& friendv)
29 {
30     int minIdx = min(i, j);
31     int maxIdx = max(i, j);
32 
33     for (auto& e : friendv) {
34         if (e.first == minIdx && e.second == maxIdx)
35             return true;
36     }
37 
38     return false;
39 }
40 
41 
42 int dfs(bool isChecked[],const vector<PII>& friendv)
43 {
44     int firstFree = -1;
45     for (int i = 0; i < n; i++) {
46         if (isChecked[i] == false){
47             firstFree = i;
48             break;
49         }
50     }
51 
52     if (-1 == firstFree)
53         return 1;
54 
55     int ret = 0;
56 
57 
58     for (int secondFree = firstFree + 1; secondFree < n; secondFree++) {
59         if (firstFree != secondFree && isChecked[firstFree] == false && isChecked[secondFree] == false 
60                 && isFriend(firstFree, secondFree, friendv)) {
61             isChecked[firstFree] = true; isChecked[secondFree] = true;
62             ret += dfs(isChecked, friendv);
63             isChecked[firstFree] = false; isChecked[secondFree] = false;
64         }
65     }
66 
67 
68     return ret;
69 }
70 
71 
72 
73 
74 int main()
75 {
76     cin >> t;
77 
78     while (t--) {
79         cin >> n >> m;
80         vector<PII> friendv;
81         bool isChecked[160] = {false};
82         while (m--) {
83             cin >> a >> b;
84             friendv.push_back({min(a,b),max(a,b)});
85         }
86         sort(friendv.begin(), friendv.end());
87         cout << dfs(isChecked, friendv)<<endl;
88     }
89 
90 
91     return 0;
92 }

View Code