#568623 (C++17) No.1261 数字集め

提出ソース

結果

問題	No.1261 数字集め
ユーザー	kmjp
提出日時	2020-10-17 20:06:26
言語	C++17 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	2,558 ms / 8,000 ms
コード長	2,061 bytes
コンパイル時間	2,393 ms
コンパイル使用メモリ	205,588 KB
実行使用メモリ	120,192 KB
最終ジャッジ日時	2024-07-21 06:05:25
合計ジャッジ時間	119,048 ms
ジャッジサーバーID （参考情報）	judge1 / judge3

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	2,188 ms 120,064 KB
testcase_01	AC	355 ms 56,192 KB
testcase_02	AC	1,147 ms 90,112 KB
testcase_03	AC	1,149 ms 90,880 KB
testcase_04	AC	647 ms 67,200 KB
testcase_05	AC	1,692 ms 112,384 KB
testcase_06	AC	742 ms 73,088 KB
testcase_07	AC	1,176 ms 93,184 KB
testcase_08	AC	486 ms 63,104 KB
testcase_09	AC	708 ms 74,752 KB
testcase_10	AC	57 ms 28,800 KB
testcase_11	AC	1,726 ms 110,848 KB
testcase_12	AC	1,507 ms 101,760 KB
testcase_13	AC	1,236 ms 94,208 KB
testcase_14	AC	58 ms 33,536 KB
testcase_15	AC	451 ms 58,752 KB
testcase_16	AC	1,094 ms 93,440 KB
testcase_17	AC	818 ms 80,128 KB
testcase_18	AC	397 ms 59,520 KB
testcase_19	AC	1,132 ms 91,008 KB
testcase_20	AC	67 ms 35,072 KB
testcase_21	AC	1,692 ms 116,096 KB
testcase_22	AC	1,686 ms 115,968 KB
testcase_23	AC	1,759 ms 116,096 KB
testcase_24	AC	1,702 ms 116,224 KB
testcase_25	AC	1,541 ms 116,224 KB
testcase_26	AC	1,565 ms 116,096 KB
testcase_27	AC	1,597 ms 116,224 KB
testcase_28	AC	1,551 ms 116,224 KB
testcase_29	AC	1,613 ms 116,224 KB
testcase_30	AC	1,676 ms 116,352 KB
testcase_31	AC	1,663 ms 116,224 KB
testcase_32	AC	1,654 ms 116,224 KB
testcase_33	AC	1,607 ms 116,224 KB
testcase_34	AC	1,593 ms 116,224 KB
testcase_35	AC	1,719 ms 116,224 KB
testcase_36	AC	1,786 ms 116,224 KB
testcase_37	AC	1,832 ms 116,224 KB
testcase_38	AC	1,726 ms 116,224 KB
testcase_39	AC	1,731 ms 116,224 KB
testcase_40	AC	1,769 ms 116,352 KB
testcase_41	AC	1,857 ms 116,352 KB
testcase_42	AC	1,775 ms 116,224 KB
testcase_43	AC	1,877 ms 116,352 KB
testcase_44	AC	1,512 ms 108,544 KB
testcase_45	AC	192 ms 44,160 KB
testcase_46	AC	591 ms 68,608 KB
testcase_47	AC	391 ms 56,576 KB
testcase_48	AC	1,219 ms 94,464 KB
testcase_49	AC	2,558 ms 106,880 KB
testcase_50	AC	1,424 ms 98,688 KB
testcase_51	AC	242 ms 46,464 KB
testcase_52	AC	38 ms 28,288 KB
testcase_53	AC	236 ms 46,720 KB
testcase_54	AC	304 ms 49,280 KB
testcase_55	AC	108 ms 37,248 KB
testcase_56	AC	158 ms 40,192 KB
testcase_57	AC	1,203 ms 91,008 KB
testcase_58	AC	210 ms 44,672 KB
testcase_59	AC	1,602 ms 109,952 KB
testcase_60	AC	267 ms 48,640 KB
testcase_61	AC	448 ms 57,984 KB
testcase_62	AC	1,418 ms 101,248 KB
testcase_63	AC	115 ms 38,784 KB
testcase_64	AC	1,788 ms 116,608 KB
testcase_65	AC	1,020 ms 84,992 KB
testcase_66	AC	781 ms 74,880 KB
testcase_67	AC	1,422 ms 99,712 KB
testcase_68	AC	797 ms 72,576 KB
testcase_69	AC	727 ms 70,784 KB
testcase_70	AC	522 ms 62,080 KB
testcase_71	AC	1,343 ms 97,536 KB
testcase_72	AC	336 ms 51,456 KB
testcase_73	AC	423 ms 55,936 KB
testcase_74	AC	153 ms 40,576 KB
testcase_75	AC	675 ms 68,736 KB
testcase_76	AC	1,209 ms 91,264 KB
testcase_77	AC	1,412 ms 102,144 KB
testcase_78	AC	41 ms 27,136 KB
testcase_79	AC	1,482 ms 104,704 KB
testcase_80	AC	53 ms 28,288 KB
testcase_81	AC	79 ms 34,432 KB
testcase_82	AC	1,015 ms 87,552 KB
testcase_83	AC	338 ms 52,992 KB
testcase_84	AC	1,818 ms 120,064 KB
testcase_85	AC	1,923 ms 120,064 KB
testcase_86	AC	1,935 ms 120,064 KB
testcase_87	AC	1,937 ms 120,064 KB
testcase_88	AC	1,869 ms 120,064 KB
testcase_89	AC	1,814 ms 120,064 KB
testcase_90	AC	1,913 ms 120,192 KB
testcase_91	AC	1,785 ms 120,064 KB
testcase_92	AC	1,816 ms 120,064 KB
testcase_93	AC	1,726 ms 120,064 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

#include <bits/stdc++.h>
using namespace std;
typedef signed long long ll;

#undef _P
#define _P(...) (void)printf(__VA_ARGS__)
#define FOR(x,to) for(x=0;x<(to);x++)
#define FORR(x,arr) for(auto& x:arr)
#define FORR2(x,y,arr) for(auto& [x,y]:arr)
#define ITR(x,c) for(__typeof(c.begin()) x=c.begin();x!=c.end();x++)
#define ALL(a) (a.begin()),(a.end())
#define ZERO(a) memset(a,0,sizeof(a))
#define MINUS(a) memset(a,0xff,sizeof(a))
//-------------------------------------------------------

int N,M;
const ll mo=1000000007;
int A[1010101];
int EN[1010101];
vector<int> R[1010101];
int Q,X,Y;
ll modpow(ll a, ll n = mo-2) {
	ll r=1;a%=mo;
	while(n) r=r*((n%2)?a:1)%mo,a=a*a%mo,n>>=1;
	return r;
}

ll hoge(ll a,ll b,ll c,ll m) {
	if(a>b) swap(a,b);
	if(b>c) swap(b,c);
	if(a>b) swap(a,b);
	if(a==c) {
		ll ret=modpow(a,M);
		//comb(m+2,2)
		return ret*((m+2)*(m+1)/2%mo)%mo;
	}
	else if(a==b) {
		ll p=modpow(a,M+1)-modpow(c,M+1)+mo;
		ll q=(mo+c-a)*(a+mo-c)%mo;
		return p*c%mo*modpow(q)%mo;
	}
	else if(b==c) {
		ll p=modpow(a,M+1)-modpow(c,M+1)+mo;
		ll q=(mo+c-a)*(a+mo-c)%mo;
		return p*a%mo*modpow(q)%mo;
	}
	ll A=modpow(a,M+2)*(mo+b-c)%mo;
	ll B=modpow(b,M+2)*(mo+c-a)%mo;
	ll C=modpow(c,M+2)*(mo+a-b)%mo;
	ll ret=A+B+C;
	ll q=(mo+a-b)*(mo+b-c)%mo*(mo+c-a)%mo;
	return ret*modpow(mo-q)%mo;
	
}

void solve() {
	int i,j,k,l,r,x,y; string s;
	
	cin>>N>>M;
	M-=3;
	for(i=2;i<=N;i++) {
		cin>>A[i];
		A[i]--;
	}
	ll ret=0;
	for(i=2;i<=N;i++) {
		for(j=2*i;j<=N;j+=i) {
			if(j==N) EN[i]=1;
			if(j<N&&N%j==0) ret+=hoge(A[i],A[j],A[N],M);
			R[j].push_back(i);
		}
	}
	ret%=mo;
	cout<<ret<<endl;
	
	cin>>Q;
	while(Q--) {
		cin>>X>>Y;
		
		if(Y<N) {
			if(EN[Y]) (ret+=hoge(A[X],A[Y],A[N],M))%=mo;
		}
		else {
			FORR(p,R[X]) (ret+=hoge(A[p],A[X],A[N],M))%=mo;
		}
		cout<<ret<<endl;
		if(Y==N) EN[X]=1;
		R[Y].push_back(X);
		
	}
	
}


int main(int argc,char** argv){
	string s;int i;
	if(argc==1) ios::sync_with_stdio(false), cin.tie(0);
	FOR(i,argc-1) s+=argv[i+1],s+='\n'; FOR(i,s.size()) ungetc(s[s.size()-1-i],stdin);
	cout.tie(0); solve(); return 0;
}

yukicoder

結果

テストケース

ソースコード