結果

問題 No.590 Replacement
ユーザー sigma425sigma425
提出日時 2017-11-06 18:58:08
言語 C++11
(gcc 11.4.0)
結果
AC  
実行時間 151 ms / 2,000 ms
コード長 3,942 bytes
コンパイル時間 2,037 ms
コンパイル使用メモリ 191,744 KB
実行使用メモリ 26,836 KB
最終ジャッジ日時 2024-11-24 04:03:43
合計ジャッジ時間 6,641 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,248 KB
testcase_02 AC 2 ms
5,248 KB
testcase_03 AC 2 ms
5,248 KB
testcase_04 AC 2 ms
5,248 KB
testcase_05 AC 1 ms
5,248 KB
testcase_06 AC 1 ms
5,248 KB
testcase_07 AC 7 ms
5,248 KB
testcase_08 AC 5 ms
5,248 KB
testcase_09 AC 8 ms
5,248 KB
testcase_10 AC 5 ms
5,248 KB
testcase_11 AC 3 ms
5,248 KB
testcase_12 AC 9 ms
5,248 KB
testcase_13 AC 39 ms
5,248 KB
testcase_14 AC 58 ms
5,320 KB
testcase_15 AC 4 ms
5,248 KB
testcase_16 AC 15 ms
5,248 KB
testcase_17 AC 25 ms
5,248 KB
testcase_18 AC 92 ms
8,724 KB
testcase_19 AC 90 ms
6,996 KB
testcase_20 AC 22 ms
5,248 KB
testcase_21 AC 113 ms
7,616 KB
testcase_22 AC 110 ms
9,832 KB
testcase_23 AC 120 ms
9,272 KB
testcase_24 AC 118 ms
8,612 KB
testcase_25 AC 119 ms
7,808 KB
testcase_26 AC 120 ms
8,032 KB
testcase_27 AC 119 ms
9,208 KB
testcase_28 AC 117 ms
7,280 KB
testcase_29 AC 119 ms
7,352 KB
testcase_30 AC 122 ms
7,020 KB
testcase_31 AC 121 ms
8,684 KB
testcase_32 AC 127 ms
9,012 KB
testcase_33 AC 1 ms
5,248 KB
testcase_34 AC 1 ms
5,248 KB
testcase_35 AC 2 ms
5,248 KB
testcase_36 AC 94 ms
11,008 KB
testcase_37 AC 94 ms
11,140 KB
testcase_38 AC 89 ms
11,140 KB
testcase_39 AC 90 ms
11,136 KB
testcase_40 AC 143 ms
12,544 KB
testcase_41 AC 143 ms
12,800 KB
testcase_42 AC 90 ms
10,720 KB
testcase_43 AC 89 ms
9,120 KB
testcase_44 AC 151 ms
26,836 KB
testcase_45 AC 95 ms
11,140 KB
testcase_46 AC 93 ms
11,008 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#define rep(i,n) for(int i=0;i<(int)(n);i++)
#define rep1(i,n) for(int i=1;i<=(int)(n);i++)
#define all(c) c.begin(),c.end()
#define pb push_back
#define fs first
#define sc second
#define show(x) cout << #x << " = " << x << endl
#define chmin(x,y) x=min(x,y)
#define chmax(x,y) x=max(x,y)
using namespace std;
template<class S,class T> ostream& operator<<(ostream& o,const pair<S,T> &p){return o<<"("<<p.fs<<","<<p.sc<<")";}
template<class T> ostream& operator<<(ostream& o,const vector<T> &vc){o<<"sz = "<<vc.size()<<endl<<"[";for(const T& v:vc) o<<v<<",";o<<"]";return o;}

template<unsigned int mod_>
struct ModInt{
	using uint = unsigned int;
	using ll = long long;
	using ull = unsigned long long;

	constexpr static uint mod = mod_;

	uint v;
	ModInt():v(0){}
	ModInt(ll v):v(normS(v%mod+mod)){}
	explicit operator bool() const {return v!=0;}
	static uint normS(const uint &x){return (x<mod)?x:x-mod;}		// [0 , 2*mod-1] -> [0 , mod-1]
	static ModInt make(const uint &x){ModInt m; m.v=x; return m;}
	ModInt operator+(const ModInt& b) const { return make(normS(v+b.v));}
	ModInt operator-(const ModInt& b) const { return make(normS(v+mod-b.v));}
	ModInt operator-() const { return make(normS(mod-v)); }
	ModInt operator*(const ModInt& b) const { return make((ull)v*b.v%mod);}
	ModInt operator/(const ModInt& b) const { return *this*b.inv();}
	ModInt& operator+=(const ModInt& b){ return *this=*this+b;}
	ModInt& operator-=(const ModInt& b){ return *this=*this-b;}
	ModInt& operator*=(const ModInt& b){ return *this=*this*b;}
	ModInt& operator/=(const ModInt& b){ return *this=*this/b;}
	ll extgcd(ll a,ll b,ll &x,ll &y) const{
		ll u[]={a,1,0},v[]={b,0,1};
		while(*v){
			ll t=*u/ *v;
			rep(i,3) swap(u[i]-=t*v[i],v[i]);
		}
		if(u[0]<0) rep(i,3) u[i]=-u[i];
		x=u[1],y=u[2];
		return u[0];
	}
	ModInt inv() const{
		ll x,y;
		extgcd(v,mod,x,y);
		return make(normS(x+mod));
	}
	bool operator==(const ModInt& b) const { return v==b.v;}
	bool operator!=(const ModInt& b) const { return v!=b.v;}
	friend istream& operator>>(istream &o,ModInt& x){
		ll tmp;
		o>>tmp;
		x=ModInt(tmp);
		return o;
	}
	friend ostream& operator<<(ostream &o,const ModInt& x){ return o<<x.v;}
};
using mint = ModInt<1000000007>;

using P = pair<int,int>;
using T = tuple<int,int,int>;

using ll = __int128;

ll extgcd(ll a,ll b,ll &x,ll &y){
	ll u[]={a,1,0},v[]={b,0,1};
	while(*v){
		ll t=*u/ *v;
		rep(i,3) swap(u[i]-=t*v[i],v[i]);
	}
	if(u[0]<0) rep(i,3) u[i]=-u[i];
	x=u[1],y=u[2];
	return u[0];
}

ll gcd(ll x,ll y){
	if(y==0) return x;
	return gcd(y,x%y);
}

int N;
vector<vector<int>> Acs,Bcs;
vector<P> A2id,B2id;	//Acs[fs][sc] = i;

int main(){
	cin>>N;
	A2id.resize(N),B2id.resize(N);
	rep(t,2){
		vector<int> p(N);
		rep(i,N) cin>>p[i],p[i]--;
		vector<bool> vis(N,false);
		rep(i,N) if(!vis[i]){
			vector<int> c;
			int v = i;
			int cid = (t==0?Acs.size():Bcs.size());
			int vid = 0;
			while(!vis[v]){
				vis[v] = 1;
				c.pb(v);
				(t==0?A2id[v]:B2id[v]) = P(cid,vid);
				v = p[v];
				vid++;
			}
			if(t==0) Acs.pb(c);
			else Bcs.pb(c);
		}
	}
	map<T,vector<P>> mp;
	rep(v,N){
		int A = A2id[v].fs, B = B2id[v].fs;
		int x = A2id[v].sc, y = B2id[v].sc;
		int a = Acs[A].size(), b = Bcs[B].size();
		int g = gcd(a,b);
		int d = ((y-x)%g+g)%g;
		mp[T{A,B,d}].pb(P(x,y));
	}
	mint ans = 0;

	for(auto it:mp){
		int A,B,d;
		tie(A,B,d) = it.fs;
		vector<P> vp = it.sc;
		ll a = Acs[A].size(), b = Bcs[B].size();
		int g = gcd(a,b);

		ll T = a*b/g;
		vector<ll> ts;
		for(P p:vp){
			ll x = p.fs, y = p.sc;
			y -= d;
			if(y<0) y += b;
			int off = x%g;
			x -= off;
			y -= off;
			ll u,v;
			extgcd(a/g,b/g,u,v);
			ll t = ( (x/g)*(b/g)*v + (y/g)*(a/g)*u ) % ((a/g)*(b/g));
			if(t<0) t += (a/g)*(b/g);
			t = t*g + off;
			ts.pb(t);
		}
		sort(all(ts));
		int K = ts.size();
		ts.pb(ts[0]+T);
		rep(i,K){
			long long L = ts[i+1]-ts[i];
			mint tmp = (mint)L*(L-1)/2;
			ans += tmp;
		}
	}
	cout<<ans<<endl;
}
0