結果

問題 No.590 Replacement
ユーザー chocoruskchocorusk
提出日時 2020-04-13 15:56:29
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 146 ms / 2,000 ms
コード長 2,292 bytes
コンパイル時間 2,063 ms
コンパイル使用メモリ 153,348 KB
実行使用メモリ 27,532 KB
最終ジャッジ日時 2024-09-25 02:20:37
合計ジャッジ時間 6,977 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 3 ms
8,668 KB
testcase_01 AC 3 ms
9,692 KB
testcase_02 AC 4 ms
8,924 KB
testcase_03 AC 4 ms
8,540 KB
testcase_04 AC 4 ms
9,052 KB
testcase_05 AC 4 ms
10,064 KB
testcase_06 AC 4 ms
9,944 KB
testcase_07 AC 9 ms
9,528 KB
testcase_08 AC 6 ms
9,964 KB
testcase_09 AC 9 ms
9,152 KB
testcase_10 AC 6 ms
9,620 KB
testcase_11 AC 5 ms
8,680 KB
testcase_12 AC 10 ms
9,768 KB
testcase_13 AC 34 ms
10,244 KB
testcase_14 AC 51 ms
11,044 KB
testcase_15 AC 6 ms
8,972 KB
testcase_16 AC 14 ms
10,184 KB
testcase_17 AC 23 ms
10,116 KB
testcase_18 AC 77 ms
13,624 KB
testcase_19 AC 78 ms
12,176 KB
testcase_20 AC 22 ms
9,556 KB
testcase_21 AC 92 ms
12,968 KB
testcase_22 AC 96 ms
14,504 KB
testcase_23 AC 99 ms
14,068 KB
testcase_24 AC 97 ms
13,120 KB
testcase_25 AC 95 ms
12,924 KB
testcase_26 AC 99 ms
12,808 KB
testcase_27 AC 97 ms
13,640 KB
testcase_28 AC 95 ms
12,092 KB
testcase_29 AC 97 ms
12,180 KB
testcase_30 AC 97 ms
12,008 KB
testcase_31 AC 99 ms
13,380 KB
testcase_32 AC 105 ms
14,100 KB
testcase_33 AC 4 ms
9,560 KB
testcase_34 AC 3 ms
8,664 KB
testcase_35 AC 3 ms
8,540 KB
testcase_36 AC 87 ms
15,468 KB
testcase_37 AC 85 ms
15,468 KB
testcase_38 AC 87 ms
15,596 KB
testcase_39 AC 88 ms
15,464 KB
testcase_40 AC 141 ms
19,412 KB
testcase_41 AC 132 ms
19,284 KB
testcase_42 AC 85 ms
14,700 KB
testcase_43 AC 86 ms
13,700 KB
testcase_44 AC 146 ms
27,532 KB
testcase_45 AC 86 ms
15,468 KB
testcase_46 AC 86 ms
15,468 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <cstdio>
#include <cstring>
#include <iostream>
#include <string>
#include <cmath>
#include <bitset>
#include <vector>
#include <map>
#include <set>
#include <queue>
#include <deque>
#include <algorithm>
#include <complex>
#include <unordered_map>
#include <unordered_set>
#include <random>
#include <cassert>
#include <fstream>
#include <utility>
#include <functional>
#include <time.h>
#include <stack>
#include <array>
#define popcount __builtin_popcount
using namespace std;
typedef long long int ll;
typedef pair<int, int> P;
const ll MOD=1e9+7;
ll inv(ll a, ll m){
    ll b=m, x=1, y=0;
    while(b>0){
        ll t=a/b;
        swap(a-=t*b, b);
        swap(x-=t*y, y);
    }
    return (x%m+m)%m;
}
int main()
{
    int n;
	cin>>n;
	int a[100010], b[100010];
	for(int i=0; i<n; i++){
		cin>>a[i];
		a[i]--;
	}
	for(int i=0; i<n; i++){
		cin>>b[i];
		b[i]--;
	}
	int p[100010], q[100010];
	fill(p, p+n, -1); fill(q, q+n, -1);
	vector<int> va[100010], vb[100010];
	int idx=0;
	for(int i=0; i<n; i++){
		if(p[i]!=-1) continue;
		int x=i;
		do{
			p[x]=idx;
			va[idx].push_back(x);
			x=a[x];
		}while(x!=i);
		idx++;
	}
	idx=0;
	for(int i=0; i<n; i++){
		if(q[i]!=-1) continue;
		int x=i;
		do{
			q[x]=idx;
			vb[idx].push_back(x);
			x=b[x];
		}while(x!=i);
		idx++;
	}
	int ra[100010], rb[100010];
	for(int i=0; i<n; i++){
		for(int j=0; j<va[i].size(); j++){
			ra[va[i][j]]=j;
		}
		for(int j=0; j<vb[i].size(); j++){
			rb[vb[i][j]]=j;
		}
	}
	map<P, vector<int>> mp;
	for(int i=0; i<n; i++){
		mp[{p[i], q[i]}].push_back(i);
	}
	auto sum=[](ll m){
		if(m%2==0) return m/2%MOD*((m-1)%MOD)%MOD;
		else return (m-1)/2%MOD*(m%MOD)%MOD;
	};
	ll ans=0;
	for(auto pr:mp){
		auto v=pr.second;
		int p1=pr.first.first, q1=pr.first.second;
		ll cp=va[p1].size(), cq=vb[q1].size();
		ll g=gcd(cp, cq);
		map<ll, vector<ll>> mp1;
		for(auto i:v){
			ll rp=ra[i], rq=rb[i];
			ll s=(rq-rp%g+g)%g;
			rq=(rq-s+cq)%cq;
			ll x=inv(cp/g, cq/g)*(((rq-rp)/g)%(cq/g)+cq/g)%(cq/g);
			mp1[s].push_back(x*cp+rp);
		}
		for(auto pr2:mp1){
			auto w=pr2.second;
			sort(w.begin(), w.end());
			w.push_back(w[0]+cp*cq/g);
			for(int i=0; i<w.size()-1; i++){
				ll m=w[i+1]-w[i];
              //cout<<m<<" "<<sum(m)<<endl;
				(ans+=sum(m))%=MOD;
			}
		}
	}
	cout<<ans<<endl;
    return 0;
}
0