洛谷:P2661
OJ: I4951
方法一: 集合 40分
代码实现:
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 | /**************************************************************** * Description: 2015_S_semi_2 set集合 40分 * Author: Alex Li * Date: 2024-09-16 13:24:28 * LastEditTime: 2024-09-20 05:51:13 ****************************************************************/ #include <iostream> #include <unordered_set> #include <vector> using namespace std; int main() { int n; cin >> n; // 存储每个学生的信息传递对象 vector<int> T(n + 1); for (int i = 1; i <= n; i++) { cin >> T[i]; } // 初始化每个学生已知的生日集合 vector<unordered_set<int>> known(n + 1); for (int i = 1; i <= n; i++) { known[i].insert(i); // 每个学生最初只知道自己的生日 } int rounds = 0; while (true) { rounds++; // 初始化本轮接收到的生日信息 vector<unordered_set<int>> received(n + 1); // 每个学生将已知的生日传递给目标学生 for (int i = 1; i <= n; i++) { for (auto birthday : known[i]) { received[T[i]].insert(birthday); // 将i号学生的已知生日信息传递给目标T[i] // 检查目标学生T[i]是否接收到自己的生日 if (birthday == T[i]) { cout << rounds << endl; return 0; // 如果接收到自己的生日,游戏结束,输出轮数 } } } // 更新每个学生已知的生日信息 for (int i = 1; i <= n; i++) { known[i].insert(received[i].begin(), received[i].end()); } } } |
方法二:并查集 100分
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 | /**************************************************************** * Description: 2015_S_semi_2 并查集 100 * Author: Alex Li * Date: 2024-09-16 18:10:45 * LastEditTime: 2024-09-16 18:18:55 ****************************************************************/ #include <iostream> #include <algorithm> // 为了使用 min 函数 using namespace std; int f[200002], d[200002], n, minn, last; // f保存祖先节点,d保存到其祖先节点的路径长。 int fa(int x){ if (f[x] != x){ // 查找时沿途更新祖先节点和路径长。 int last = f[x]; // 记录父节点(会在递归中被更新)。 f[x] = fa(f[x]); // 更新祖先节点。 d[x] += d[last]; // 更新路径长(原来连在父节点上)。 } return f[x]; } void check(int a, int b){ int x = fa(a), y = fa(b); // 查找祖先节点。 if (x != y) { f[x] = y; d[a] = d[b] + 1; // 若不相连,则连接两点,更新父节点和路径长。 } else { minn = min(minn, d[a] + d[b] + 1); // 若已连接,则更新最小环长度。 } return; } int main(){ int i, t; cin >> n; for (i = 1; i <= n; i++) f[i] = i; // 祖先节点初始化为自己,路径长为0。 minn = 0x7777777; for (i = 1; i <= n; i++){ cin >> t; check(i, t); // 检查当前两点是否已有边相连接。 } cout << minn << endl; return 0; } |