#89023 (C++14) No.368 LCM of K-products

提出ソース

結果

問題	No.368 LCM of K-products
ユーザー	torus711
提出日時	2016-04-29 23:07:32
言語	C++14 (gcc 13.2.0 + boost 1.83.0)
結果	AC
実行時間	327 ms / 2,000 ms
コード長	3,652 bytes
コンパイル時間	1,436 ms
コンパイル使用メモリ	125,224 KB
実行使用メモリ	4,348 KB
最終ジャッジ日時	2023-10-23 21:10:37
合計ジャッジ時間	3,117 ms
ジャッジサーバーID （参考情報）	judge13 / judge14

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

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	28 ms 4,348 KB
testcase_01	AC	93 ms 4,348 KB
testcase_02	AC	3 ms 4,348 KB
testcase_03	AC	30 ms 4,348 KB
testcase_04	AC	9 ms 4,348 KB
testcase_05	AC	327 ms 4,348 KB
testcase_06	AC	2 ms 4,348 KB
testcase_07	AC	2 ms 4,348 KB
testcase_08	AC	2 ms 4,348 KB
testcase_09	AC	2 ms 4,348 KB
testcase_10	AC	1 ms 4,348 KB
testcase_11	AC	2 ms 4,348 KB
testcase_12	AC	1 ms 4,348 KB
testcase_13	AC	21 ms 4,348 KB
testcase_14	AC	33 ms 4,348 KB
testcase_15	AC	36 ms 4,348 KB
testcase_16	AC	30 ms 4,348 KB
testcase_17	AC	27 ms 4,348 KB
testcase_18	AC	37 ms 4,348 KB
testcase_19	AC	10 ms 4,348 KB
testcase_20	AC	27 ms 4,348 KB
testcase_21	AC	8 ms 4,348 KB
testcase_22	AC	36 ms 4,348 KB
testcase_23	AC	2 ms 4,348 KB
testcase_24	AC	2 ms 4,348 KB
testcase_25	AC	2 ms 4,348 KB
testcase_26	AC	2 ms 4,348 KB
testcase_27	AC	2 ms 4,348 KB
testcase_28	AC	1 ms 4,348 KB
testcase_29	AC	2 ms 4,348 KB
testcase_30	AC	1 ms 4,348 KB
testcase_31	AC	2 ms 4,348 KB
testcase_32	AC	1 ms 4,348 KB
testcase_33	AC	15 ms 4,348 KB
testcase_34	AC	2 ms 4,348 KB

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

ソースコード

raw source code

#include <iostream>
#include <iomanip>
#include <sstream>
#include <vector>
#include <string>
#include <set>
#include <unordered_set>
#include <map>
#include <unordered_map>
#include <stack>
#include <queue>
#include <deque>
#include <algorithm>
#include <functional>
#include <iterator>
#include <limits>
#include <numeric>
#include <utility>
#include <cmath>
#include <cassert>
#include <cstdio>

using namespace std;
using namespace placeholders;

using LL = long long;
using ULL = unsigned long long;
using VI = vector< int >;
using VVI = vector< vector< int > >;
using VS = vector< string >;
using SS = stringstream;
using PII = pair< int, int >;
using VPII = vector< pair< int, int > >;
template < typename T = int > using VT = vector< T >;
template < typename T = int > using VVT = vector< vector< T > >;
template < typename T = int > using LIM = numeric_limits< T >;

template < typename T > inline istream& operator>>( istream &s, vector< T > &v ){ for ( T &t : v ) { s >> t; } return s; }
template < typename T > inline ostream& operator<<( ostream &s, const vector< T > &v ){ for ( int i = 0; i < int( v.size() ); ++i ){ s << ( " " + !i ) << v[i]; } return s; }
template < typename T > inline T fromString( const string &s ) { T res; istringstream iss( s ); iss >> res; return res; }
template < typename T > inline string toString( const T &a ) { ostringstream oss; oss << a; return oss.str(); }

#define REP2( i, n ) REP3( i, 0, n )
#define REP3( i, m, n ) for ( int i = ( int )( m ); i < ( int )( n ); ++i )
#define GET_REP( a, b, c, F, ... ) F
#define REP( ... ) GET_REP( __VA_ARGS__, REP3, REP2 )( __VA_ARGS__ )
#define FOR( e, c ) for ( auto &&e : c )
#define ALL( c ) begin( c ), end( c )
#define AALL( a, t ) ( t* )a, ( t* )a + sizeof( a ) / sizeof( t )
#define DRANGE( c, p ) ( c ).begin(), ( c ).begin() + ( p ), ( c ).end()

#define SZ( v ) ( (int)( v ).size() )
#define EXIST( c, e ) ( ( c ).find( e ) != ( c ).end() )

template < typename T > inline bool chmin( T &a, const T &b ){ if ( b < a ) { a = b; return true; } return false; }
template < typename T > inline bool chmax( T &a, const T &b ){ if ( a < b ) { a = b; return true; } return false; }

#define PB push_back
#define EM emplace
#define EB emplace_back
#define BI back_inserter

#define MP make_pair
#define fst first
#define snd second

#define DUMP( x ) cerr << #x << " = " << ( x ) << endl

constexpr int MOD = 1'000'000'007;

// 素因数分解 O( √N )
vector< long long > primeFactorization( long long N )
{
	if ( N == 1 )
	{
		return vector< long long >();
	}

	vector< long long > result;
	for ( long long p = 2; p * p <= N; p++ )
	{
		while ( !( N % p ) )
		{
			result.push_back( p );
			N /= p;
		}
	}
	if ( N != 1 )
	{
		result.push_back( N );
	}

	return result;
}

// a^x を mod で求める
// 反復二乗法
// O( log x )
long long mod_pow( long long a, long long x, long long mod )
{
	a %= mod;

	long long res = 1;
	for ( ; x; x >>= 1, ( a *= a ) %= mod )
	{
		if ( x & 1 )
		{
			( res *= a ) %= mod;
		}
	}
	return res;
}

int main()
{
	cin.tie( 0 );
	ios::sync_with_stdio( false );

	int N, K;
	cin >> N >> K;

	VI A( N );
	cin >> A;

	map< int, VI > exps;
	FOR( a, A )
	{
		const auto factors = primeFactorization( a );

		map< int, int > counts;
		FOR( p, factors )
		{
			++counts[p];
		}

		FOR( p, counts )
		{
			exps[ p.fst ].PB( p.snd );
		}
	}

	LL res = 1;
	FOR( p, exps )
	{
		const int prime = p.fst;
		VI exp = p.snd;

		sort( ALL( exp ), greater< int >() );

		const int e = accumulate( begin( exp ), begin( exp ) + min( SZ( exp ), K ), 0 );
		( res *= mod_pow( prime, e, MOD ) ) %= MOD;
	}

	cout << res << endl;

	return 0;
}

yukicoder

結果

テストケース

ソースコード