P4979.第2题-多约束条件下的元素匹配统计
题目内容
给定三个长度为 n 的数组 {a1,a2,…,an}、{b1,b2,…,bn} 和 {c1,c2,…,cn}。这里 cj 表示对数组 b 的一个下标。
请你统计满足以下条件的有序对 (i,j) 的数量:
- 1≤i≤j≤n;
- ai=bcj。
输入描述
每个测试文件均包含多组测试数据。第一行输入一个整数 T(1≤T≤2×105)代表数据组数;除此之外,保证单个测试文件的 n 之和不超过 2×105。
每组测试数据的输入格式如下:
- 第一行输入一个整数 n(1≤n≤105);
- 第二行输入 n 个整数 a1,a2,…,an(1≤ai≤109);
- 第三行输入 n 个整数 b1,b2,…,bn(1≤bi≤109);
- 第四行输入 n 个整数 c1,c2,…,cn(1≤cj≤n)。
输出描述
对于每一组测试数据,新起一行输出一个整数,表示满足条件的有序对数量。
样例1
输入
2
5
1 2 1 2 1
2 1 3 1 2
2 1 5 4 3
3
7 7 7
1 2 7
3 3 3
输出
5
6