题意:给你n个数代表序列a(1~n),然后给你m个数代表序列b(1~m), 问有多少种函数能满足f(i)=b[f(ai)],输出所有的可能结果,答案余1e9+7。
思路:这个题目比赛的时候看到一看要余1e9+7第一感觉就是这个题目是一个组合数学的题目,然后我就在自己一直凑样例,而没有对题目给的函数式的关系进行进一步的推导。导致自己没有什么思路。
赛后看别人的题解发现这个题目是要去找循环节来做,例如第一组样例
f(0) = b[f(1)] = b[b[f(0)]]
f(1) = b[f(0)] = b[b[f(1)]]
f(2) = b[f(2)]
可以发现每一个式子都可以表达为
f(i) = b[f(a[i])] = b[b[f(a[a[i]])]] = b…b[f(i)]
我们发现a中的l个数构成了循环群,而只有当j为l的因子时才满足上述函数,然后我们就可以把序列a当中的所有环找出来(包括自环),把序列b当中的所有环都找出来,如果b中的一个环中包含数字的个数是a中的一个环的数字个数的因数说明这两个环是可以构成一个循环的,满足条件。
a中每种循环节都是单独的,而a中循环节对应的b中的循环节则要相加。
#include <algorithm>
#include <cmath>
#include <cstdio>
#include <cstring>
#include <fstream>
#include <iostream>
#include <map>
#include <queue>
#include <set>
#include <stack>
#include <string>
#include <vector>
#include <sstream>
#include <list>
#define MAXN 100100
#define INF 10000000
#define MOD 1000000007
#define LL long long
#define pi acos(-1.0)
using namespace std;
int arr[MAXN];
int brr[MAXN];
bool vis[MAXN];
void find_circle(int *a, int n, vector<int> &vec) {
memset(vis, false, sizeof(vis));
for (int i = 0; i < n; ++i) {
if (!vis[i]) {
int now = a[i], len = 0;
while (!vis[now]) {
len++;
vis[now] = true;
now = a[now];
}
vec.push_back(len);
}
}
}
int main() {
std::ios::sync_with_stdio(false);
int n, m;
int kase = 0;
while (cin >> n >> m) {
for (int i = 0; i < n; ++i)
cin >> arr[i];
for (int i = 0; i < m; ++i)
cin >> brr[i];
vector<int> vec1;
vector<int> vec2;
find_circle(arr, n, vec1);
find_circle(brr, m, vec2);
LL ans = 1;
//cout << vec1.size() << " " << vec2.size() << endl;
for (int i = 0; i < vec1.size(); ++i) {
LL cnt = 0;
for (int j = 0; j < vec2.size(); ++j) {
if (vec1[i] % vec2[j] == 0) {
cnt += vec2[j];
cnt %= MOD;
}
}
ans *= cnt;
ans %= MOD;
}
cout << "Case #" << ++kase << ": ";
cout << ans << endl;
}
return 0;
}