#565626 (C++17) No.1255 ハイレーツ・オブ・ボリビアン

• 問題: No.1255 ハイレーツ・オブ・ボリビアン
• ユーザー: 沙耶花
• 提出日時: 2020-10-10 15:58:34
• 言語: C++17 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 1,116 ms / 2,000 ms
• コード長: 931 bytes
• コンパイル時間: 2,215 ms
• コンパイル使用メモリ: 211,608 KB
• 実行使用メモリ: 5,824 KB
• 最終ジャッジ日時: 2024-07-20 16:01:52
• 合計ジャッジ時間: 7,251 ms
• ジャッジサーバーID （参考情報）: judge3 / judge2

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 5,248 KB
testcase_01	AC	2 ms; 5,376 KB
testcase_02	AC	2 ms; 5,376 KB
testcase_03	AC	2 ms; 5,376 KB
testcase_04	AC	2 ms; 5,376 KB
testcase_05	AC	6 ms; 5,376 KB
testcase_06	AC	5 ms; 5,376 KB
testcase_07	AC	6 ms; 5,376 KB
testcase_08	AC	485 ms; 5,576 KB
testcase_09	AC	517 ms; 5,824 KB
testcase_10	AC	508 ms; 5,816 KB
testcase_11	AC	495 ms; 5,584 KB
testcase_12	AC	526 ms; 5,580 KB
testcase_13	AC	46 ms; 5,752 KB
testcase_14	AC	1,116 ms; 5,704 KB
testcase_15	AC	485 ms; 5,824 KB

ソースコード

#include <stdio.h>
#include <bits/stdc++.h>
#include <atcoder/math>
using namespace atcoder;
using namespace std;
#define rep(i,n) for (int i = 0; i < (n); ++i)
#define Inf 1000000000000000000

long long modlog(long long a,long long b,long long m){
	long long sq;
	{
		long long ng = 0LL,ok = m;
		while(ok-ng>1){
			long long mid = (ok+ng)/2;
			if(mid*mid>=m){
				ok = mid;
			}
			else ng = mid;
		}
		sq = ok+1;
	}
	
	unordered_map<long long,long long> mp;
	long long t = 1;
	rep(i,sq){
		if(!mp.count(t)){
			mp[t] = i;
		}
		t *= a;
		t %= m;
	}
	t = inv_mod(t,m);
	long long now = 1;
	rep(i,sq){
		long long temp = b * now;
		temp %= m;
		if(mp.count(temp)){
			return i*sq + mp[temp];
		}
		now *= t;
		now %= m;
	}
	
	return -1;
	
}

int main(){	

	int t;
	cin>>t;
	
	rep(_,t){
		long long n;
		cin>>n;
		
		long long m = 2*n-1;
		if(n==1)cout<<1<<endl;
		else cout<<modlog(2,inv_mod(2,m),m)+1<<endl;
	}
	
    return 0;
}