#102074 (C++14) No.389 ロジックパズルの組み合わせ

提出ソース

結果

問題	No.389 ロジックパズルの組み合わせ
ユーザー	torus711
提出日時	2016-07-08 22:36:19
言語	C++14 (gcc 13.2.0 + boost 1.83.0)
結果	AC
実行時間	75 ms / 2,000 ms
コード長	4,792 bytes
コンパイル時間	913 ms
コンパイル使用メモリ	105,396 KB
実行使用メモリ	13,420 KB
最終ジャッジ日時	2024-04-21 06:02:30
合計ジャッジ時間	6,938 ms
ジャッジサーバーID （参考情報）	judge1 / judge5

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

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	35 ms 11,264 KB
testcase_01	AC	35 ms 11,184 KB
testcase_02	AC	37 ms 11,188 KB
testcase_03	AC	36 ms 11,136 KB
testcase_04	AC	36 ms 11,136 KB
testcase_05	AC	37 ms 11,264 KB
testcase_06	AC	36 ms 11,352 KB
testcase_07	AC	37 ms 11,136 KB
testcase_08	AC	35 ms 11,124 KB
testcase_09	AC	39 ms 11,648 KB
testcase_10	AC	39 ms 11,488 KB
testcase_11	AC	42 ms 11,520 KB
testcase_12	AC	39 ms 11,264 KB
testcase_13	AC	50 ms 12,268 KB
testcase_14	AC	41 ms 11,520 KB
testcase_15	AC	37 ms 11,520 KB
testcase_16	AC	39 ms 11,444 KB
testcase_17	AC	40 ms 11,520 KB
testcase_18	AC	64 ms 13,416 KB
testcase_19	AC	37 ms 11,008 KB
testcase_20	AC	37 ms 11,136 KB
testcase_21	AC	37 ms 11,008 KB
testcase_22	AC	37 ms 11,264 KB
testcase_23	AC	35 ms 11,136 KB
testcase_24	AC	36 ms 11,136 KB
testcase_25	AC	37 ms 11,264 KB
testcase_26	AC	35 ms 11,136 KB
testcase_27	AC	35 ms 11,136 KB
testcase_28	AC	38 ms 11,136 KB
testcase_29	AC	37 ms 11,264 KB
testcase_30	AC	35 ms 11,132 KB
testcase_31	AC	37 ms 11,264 KB
testcase_32	AC	36 ms 11,124 KB
testcase_33	AC	35 ms 11,136 KB
testcase_34	AC	35 ms 11,344 KB
testcase_35	AC	34 ms 11,196 KB
testcase_36	AC	36 ms 11,136 KB
testcase_37	AC	36 ms 11,136 KB
testcase_38	AC	36 ms 11,264 KB
testcase_39	AC	35 ms 11,252 KB
testcase_40	AC	36 ms 11,136 KB
testcase_41	AC	35 ms 11,188 KB
testcase_42	AC	37 ms 11,136 KB
testcase_43	AC	37 ms 11,008 KB
testcase_44	AC	36 ms 11,136 KB
testcase_45	AC	35 ms 11,124 KB
testcase_46	AC	36 ms 11,296 KB
testcase_47	AC	37 ms 11,132 KB
testcase_48	AC	39 ms 11,196 KB
testcase_49	AC	65 ms 13,296 KB
testcase_50	AC	39 ms 11,264 KB
testcase_51	AC	38 ms 11,264 KB
testcase_52	AC	41 ms 11,392 KB
testcase_53	AC	38 ms 11,264 KB
testcase_54	AC	41 ms 11,520 KB
testcase_55	AC	35 ms 11,196 KB
testcase_56	AC	36 ms 11,392 KB
testcase_57	AC	47 ms 12,268 KB
testcase_58	AC	59 ms 13,416 KB
testcase_59	AC	54 ms 12,268 KB
testcase_60	AC	45 ms 11,776 KB
testcase_61	AC	75 ms 13,420 KB
testcase_62	AC	45 ms 11,776 KB
testcase_63	AC	59 ms 13,420 KB
testcase_64	AC	43 ms 11,564 KB
testcase_65	AC	42 ms 11,460 KB
testcase_66	AC	44 ms 11,776 KB
testcase_67	AC	42 ms 11,644 KB
testcase_68	AC	43 ms 11,776 KB
testcase_69	AC	36 ms 11,184 KB
testcase_70	AC	36 ms 11,120 KB
testcase_71	AC	36 ms 11,136 KB
testcase_72	AC	36 ms 11,296 KB
testcase_73	AC	35 ms 11,136 KB
testcase_74	AC	36 ms 11,136 KB
testcase_75	AC	36 ms 11,176 KB
testcase_76	AC	36 ms 11,136 KB
testcase_77	AC	36 ms 11,264 KB
testcase_78	AC	34 ms 11,124 KB
testcase_79	AC	36 ms 11,136 KB
testcase_80	AC	35 ms 11,264 KB
testcase_81	AC	37 ms 11,196 KB
testcase_82	AC	38 ms 11,120 KB
testcase_83	AC	37 ms 11,008 KB
testcase_84	AC	37 ms 11,136 KB
testcase_85	AC	37 ms 11,264 KB
testcase_86	AC	38 ms 11,136 KB
testcase_87	AC	37 ms 11,136 KB
testcase_88	AC	35 ms 11,136 KB
testcase_89	AC	44 ms 11,888 KB
testcase_90	AC	51 ms 12,268 KB
testcase_91	AC	62 ms 13,420 KB
testcase_92	AC	38 ms 11,648 KB
testcase_93	AC	51 ms 12,396 KB
testcase_94	AC	36 ms 11,188 KB
testcase_95	AC	55 ms 12,272 KB
testcase_96	AC	53 ms 12,268 KB
testcase_97	AC	50 ms 12,268 KB
testcase_98	AC	43 ms 11,752 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;

// 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;
}


// p が素数のとき、p を法とする剰余体での逆元を求める
// Fermat の小定理を利用
// a^{ p - 1 } \equiv 1 ( mod p )
// a^{ p - 2 } \equiv a^{-1} ( mod p )
// incluide : mod_pow
int mod_inverse( long long a, long long p )
{
	return mod_pow( a, p - 2, p );
}

// 素数を法とする剰余体での n! を求める
class modFact
{
private:
	const int MAX_N, MOD;
	vector<int> fact;

public:
	modFact( const int n, const int mod ) : MAX_N( n ), MOD( mod ), fact( min( MAX_N + 1, MOD ) )
	{
		fact[0] = 1;
		for ( int i = 1; i < (int)fact.size(); ++i )
		{
			fact[i] = 1LL * fact[ i - 1 ] * i % MOD;
		}
		return;
	}

	int operator()( const int n )
	{
		int e;
		return operator()( n, e );
	}

	int operator()( const int n, int &e )
	{
		e = 0;
		if ( n == 0 )
		{
			return 1;
		}

		const long long res = operator()( n / MOD, e );
		e += n / MOD;

		if ( n / MOD % 2 != 0 )
		{
			return res * ( MOD - fact[ n % MOD ] ) % MOD;
		}
		return res * fact[ n % MOD ] % MOD;
	}
};
// modfact( max_n, mod )
// ()( n )
// ()( n, e )

// nCr
// include : modFact, mod_inverse
class modComb
{
private:
	const int MOD;
	modFact mod_fact;

public:
	modComb( const int n, const int mod ) : MOD( mod ), mod_fact( n, mod ) {};

	int operator()( const int n, const int r )
	{
		if ( n < 0 || r < 0 || n < r )
		{
			return 0;
		}

		int e1, e2, e3;
		long long a1 = mod_fact( n, e1 ), a2 = mod_fact( r, e2 ), a3 = mod_fact( n - r, e3 );
		if ( e1 > e2 + e3 )
		{
			return 0;
		}
		return a1 * mod_inverse( a2 * a3 % MOD, MOD ) % MOD;
	}
};
// modComb( n, mod )
// ()( n, r )

modComb nCr( 2000000, MOD );

int main()
{
	cin.tie( 0 );
	ios::sync_with_stdio( false );
	cout << setprecision( 12 ) << fixed;

	int N;
	scanf( "%d", &N );

	VI A;
	for ( int a; ~scanf( "%d", &a ); A.PB( a ) );

	if ( A[0] == 0 )
	{
		printf( "%d\n", 1 ) ;
		return 0;
	}

	const int blank = N - accumulate( ALL( A ), 0 );
	const int choice = SZ( A ) + 1;
	const int rest = blank - SZ( A ) + 1;

	if ( rest < 0 )
	{
		puts( "NA" );
		return 0;
	}

	printf( "%d\n", int( nCr( choice + rest - 1, rest ) ) );

	return 0;
}

yukicoder

結果

テストケース

ソースコード