#1179791 (C++23) No.3651 K-th Sum of Divisors

• 問題: No.3651 K-th Sum of Divisors
• ユーザー: askr58
• 提出日時: 2026-08-28 21:34:36
• 言語: C++23 (gcc 15.2.0 + boost 1.90.0)
• 結果: AC
• 実行時間: 555 ms / 2,000 ms
• コード長: 3,595 bytes
• コンパイル時間: 3,386 ms
• コンパイル使用メモリ: 332,492 KB
• 実行使用メモリ: 75,552 KB
• 最終ジャッジ日時: 2026-08-28 21:35:18
• 合計ジャッジ時間: 18,762 ms
• ジャッジサーバーID （参考情報）: judge3_0 / judge2_0

ファイルパターン	結果
sample	AC * 3
other	AC * 55

ソースコード

#include <iostream>
#include <random>
#include <chrono>
#include <iomanip>
#include <set>
#include <map>
#include <queue>
#include <deque>
#include <string>
#include <stack>
#include <ranges>
#include <algorithm>
#include <vector>
using namespace std;
using ll=long long;

#include <atcoder/all>
using mint=atcoder::modint998244353;

ostream& operator<<(ostream& os,const mint& x){
	os<<x.val();
	return os;
}
istream& operator>>(istream& is,mint& x){
	int t;
	is>>t;
	x=t;
	return is;
}

template <typename S,typename T>
ostream& operator<<(ostream& os,const pair<S,T>& p);
template <typename S,typename T>
istream& operator>>(istream& is,pair<S,T>& p);
template <typename T,size_t n>
ostream& operator<<(ostream& os,const array<T,n>& arr);
template <typename T,size_t n>
istream& operator>>(istream& is,array<T,n>& arr);
template <typename T>
ostream& operator<<(ostream& os,const vector<T>& vec);
template <typename T>
istream& operator>>(istream& is,vector<T>& vec);

template <typename S,typename T>
ostream& operator<<(ostream& os,const pair<S,T>& p){
	os<<p.first<<" "<<p.second;
	return os;
}
template <typename S,typename T>
istream& operator>>(istream& is,pair<S,T>& p){
	is>>p.first>>p.second;
	return is;
}

template <typename T,size_t n>
ostream& operator<<(ostream& os,const array<T,n>& arr){
	for(int i=0;i<n;i++)os<<arr[i]<<(i+1==n?"":" ");
	return os;
}
template <typename T,size_t n>
istream& operator>>(istream& is,array<T,n>& arr){
	for(int i=0;i<n;i++)is>>arr[i];
	return is;
}
template <typename T>
ostream& operator<<(ostream& os,const vector<T>& vec){
	for(int i=0;i<vec.size();i++)os<<vec[i]<<(i+1==vec.size()?"":" ");
	return os;
}
template <typename T>
istream& operator>>(istream& is,vector<T>& vec){
	for(int i=0;i<vec.size();i++)is>>vec[i];
	return is;
}

template<class... Vecs>
void input_vec(Vecs&... vs) {
    const auto n = get<0>(tie(vs...)).size();

    for (size_t i = 0; i < n; ++i)
        ((cin >> vs[i]), ...);
}

template <typename T>
vector<T> make_unique(vector<T> vec){
	ranges::sort(vec);
	vec.erase(unique(vec.begin(),vec.end()),vec.end());
	return vec;
}

template <typename T, typename Comp = ranges::less, typename Proj = identity>
pair<vector<int>,vector<int>> make_rank(const vector<T>& vec, Comp comp = {}, Proj proj = {}) {
    int n = vec.size();
    vector<int> argsort(n);
    iota(argsort.begin(), argsort.end(), 0);

    ranges::stable_sort(argsort, comp, [&](int i) -> decltype(auto) {
        return invoke(proj, vec[i]);
    });

	vector<int> rank(n);
	for(int i=0;i<n;i++)rank[argsort[i]]=i;
    return make_pair(rank,argsort);
}

using vl=vector<ll>;
using vvl=vector<vector<ll>>;
using vvvl=vector<vector<vector<ll>>>; 
using vi=vector<int>;
using vvi=vector<vector<int>>;
using vvvi=vector<vector<vector<int>>>;

pair<vector<ll>,vector<ll>> linear_sieve(int n){
	vector<ll> primes,lpf(n+1,-1);
	for(ll i=2;i<=n;i++){
		if(lpf[i]==-1){
			lpf[i]=i;
			primes.push_back(i);
		}
		for(ll p:primes){
			if(lpf[i]<p||p*i>n)break;
			lpf[p*i]=p;
		}
	}
	return make_pair(primes,lpf);
}
int main(){
	cin.tie(nullptr);
	ios::sync_with_stdio(false);
	cout<<fixed<<setprecision(10);
	ll n;
	cin>>n;
	ll mod=100003;
	ll m=max(n,mod);
	auto[primes,lpf]=linear_sieve(m+1);
	vl a(m+1);
	for(ll i=1;i<=m;i++){
		ll x=i;
		a[i]=1;
		while(x>1){
			ll s=1;
			ll p=lpf[x];
			ll t=1;
			while(x%p==0){
				x/=p;
				t*=p;
				s+=t;
			}
			a[i]=a[i]*s%mod;
		}
	}
	ll k;
	cin>>k;
	ll ans=n;
	k--;
	while(k>0){
		if(k&1){
			ans=a[ans];
		}
		vl na(m+1);
		for(ll i=1;i<=m;i++)na[i]=a[a[i]];
		a=move(na);
		k>>=1;
	}
	cout<<ans<<endl;
}