A13287 | Infinite Path
时间限制1s
内存限制256MB
通过 / 提交0/0
题目描述
You are given a colored permutation $p_1, p_2, \dots, p_n$ . The $i$ -th element of the permutation has color $c_i$ .
Let's define an infinite path as infinite sequence $i, p[i], p[p[i]], p[p[p[i]]] \dots$ where all elements have same color ( $c[i] = c[p[i]] = c[p[p[i]]] = \dots$ ).
We can also define a multiplication of permutations $a$ and $b$ as permutation $c = a \times b$ where $c[i] = b[a[i]]$ . Moreover, we can define a power $k$ of permutation $p$ as $p^k=\underbrace{p \times p \times \dots \times p}_{k \text{ times}}$ .
Find the minimum $k > 0$ such that $p^k$ has at least one infinite path (i.e. there is a position $i$ in $p^k$ such that the sequence starting from $i$ is an infinite path).
It can be proved that the answer always exists.
Let's define an infinite path as infinite sequence $i, p[i], p[p[i]], p[p[p[i]]] \dots$ where all elements have same color ( $c[i] = c[p[i]] = c[p[p[i]]] = \dots$ ).
We can also define a multiplication of permutations $a$ and $b$ as permutation $c = a \times b$ where $c[i] = b[a[i]]$ . Moreover, we can define a power $k$ of permutation $p$ as $p^k=\underbrace{p \times p \times \dots \times p}_{k \text{ times}}$ .
Find the minimum $k > 0$ such that $p^k$ has at least one infinite path (i.e. there is a position $i$ in $p^k$ such that the sequence starting from $i$ is an infinite path).
It can be proved that the answer always exists.
输入格式
The first line contains single integer $T$ ( $1 \le T \le 10^4$ ) — the number of test cases.
Next $3T$ lines contain test cases — one per three lines. The first line contains single integer $n$ ( $1 \le n \le 2 \cdot 10^5$ ) — the size of the permutation.
The second line contains $n$ integers $p_1, p_2, \dots, p_n$ ( $1 \le p_i \le n$ , $p_i \neq p_j$ for $i \neq j$ ) — the permutation $p$ .
The third line contains $n$ integers $c_1, c_2, \dots, c_n$ ( $1 \le c_i \le n$ ) — the colors of elements of the permutation.
It is guaranteed that the total sum of $n$ doesn't exceed $2 \cdot 10^5$ .
Next $3T$ lines contain test cases — one per three lines. The first line contains single integer $n$ ( $1 \le n \le 2 \cdot 10^5$ ) — the size of the permutation.
The second line contains $n$ integers $p_1, p_2, \dots, p_n$ ( $1 \le p_i \le n$ , $p_i \neq p_j$ for $i \neq j$ ) — the permutation $p$ .
The third line contains $n$ integers $c_1, c_2, \dots, c_n$ ( $1 \le c_i \le n$ ) — the colors of elements of the permutation.
It is guaranteed that the total sum of $n$ doesn't exceed $2 \cdot 10^5$ .
输出格式
Print $T$ integers — one per test case. For each test case print minimum $k > 0$ such that $p^k$ has at least one infinite path.
输入输出样例
输入 #1
3 4 1 3 4 2 1 2 2 3 5 2 3 4 5 1 1 2 3 4 5 8 7 4 5 6 1 8 3 2 5 3 6 4 7 5 8 4
输出 #1
1 5 2
In the first test case, $p^1 = p = [1, 3, 4, 2]$ and the sequence starting from $1$ : $1, p[1] = 1, \dots$ is an infinite path.
In the second test case, $p^5 = [1, 2, 3, 4, 5]$ and it obviously contains several infinite paths.
In the third test case, $p^2 = [3, 6, 1, 8, 7, 2, 5, 4]$ and the sequence starting from $4$ : $4, p^2[4]=8, p^2[8]=4, \dots$ is an infinite path since $c_4 = c_8 = 4$ .
In the second test case, $p^5 = [1, 2, 3, 4, 5]$ and it obviously contains several infinite paths.
In the third test case, $p^2 = [3, 6, 1, 8, 7, 2, 5, 4]$ and the sequence starting from $4$ : $4, p^2[4]=8, p^2[8]=4, \dots$ is an infinite path since $c_4 = c_8 = 4$ .
C++ 编辑器
输入
输出
可保存默认模板;新题优先使用已保存模板。
当前快捷键仅展示,暂不支持修改。
- 撤销
Ctrl / ⌘ + Z - 重做
Ctrl / ⌘ + Y - 查找
Ctrl / ⌘ + F - 全选
Ctrl / ⌘ + A - 复制
Ctrl / ⌘ + C - 剪切
Ctrl / ⌘ + X - 粘贴
Ctrl / ⌘ + V - 自动排版
工具栏排版按钮 - 草稿保存
编辑时自动保存到本机
历史
提交记录
状态说明时间源码
AI
作答助手
你好,我是作答助手。可以问思路、复杂度、样例含义或代码报错原因;不会直接给出完整 AC 代码。
确定要清空代码吗?
提交通过
评测结果:Accepted