PolandBall and Forest
作者:互联网
PolandBall lives in a forest with his family. There are some trees in the forest. Trees are undirected acyclic graphs with k vertices and k - 1 edges, where k is some integer. Note that one vertex is a valid tree.
There is exactly one relative living in each vertex of each tree, they have unique ids from 1 to n. For each Ball i we know the id of its most distant relative living on the same tree. If there are several such vertices, we only know the value of the one with smallest id among those.
How many trees are there in the forest?
InputThe first line contains single integer n (1 ≤ n ≤ 104) — the number of Balls living in the forest.
The second line contains a sequence p1, p2, ..., pn of length n, where (1 ≤ pi ≤ n) holds and pi denotes the most distant from Ball i relative living on the same tree. If there are several most distant relatives living on the same tree, pi is the id of one with the smallest id.
It's guaranteed that the sequence p corresponds to some valid forest.
Hacking: To hack someone, you should provide a correct forest as a test. The sequence p will be calculated according to the forest and given to the solution you try to hack as input. Use the following format:
In the first line, output the integer n (1 ≤ n ≤ 104) — the number of Balls and the integer m (0 ≤ m < n) — the total number of edges in the forest. Then m lines should follow. The i-th of them should contain two integers ai and bi and represent an edge between vertices in which relatives ai and bi live. For example, the first sample is written as follows:
Output
5 3
1 2
3 4
4 5
You should output the number of trees in the forest where PolandBall lives.
InteractionFrom the technical side, this problem is interactive. However, it should not affect you (except hacking) since there is no interaction.
Examples input Copy5output Copy
2 1 5 3 3
2input Copy
1output Copy
1
1Note
In the first sample testcase, possible forest is: 1-2 3-4-5.
There are 2 trees overall.
In the second sample testcase, the only possible graph is one vertex and no edges. Therefore, there is only one tree.
并查集模版?
#include <iostream> #include <vector> #include <algorithm> #include <string> #include <set> #include <queue> #include <map> #include <sstream> #include <cstdio> #include <cstring> #include <numeric> #include <cmath> #include <unordered_set> #include <unordered_map> //#include <xfunctional> #define ll long long #define mod 998244353 using namespace std; int dir[4][2] = { {0,1},{0,-1},{-1,0},{1,0} }; const long long inf = 0x7f7f7f7f7f7f7f7f; const int INT = 0x3f3f3f3f; int n; vector<int> a; int finds(int x) { return x == a[x] ? x : finds(a[x]); } int main() { int n; cin >> n; a.resize(n + 1); for (int i = 1; i <= n; i++) { a[i] = i; } for (int i = 1; i <= n; i++) { int m, t1, t2; cin >> m; t1 = finds(i); t2 = finds(m); if (t2 != t1) a[t2] = t1; } int ans = 0; for (int i = 1; i <= n; i++) { if (i == a[i]) ans++; } cout << ans; return 0; }
标签:living,int,tree,should,Forest,PolandBall,include,forest 来源: https://www.cnblogs.com/dealer/p/12369990.html