結果

問題 No.389 ロジックパズルの組み合わせ
コンテスト
ユーザー IL_msta
提出日時 2017-01-29 12:12:29
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
AC  
実行時間 36 ms / 2,000 ms
コード長 2,045 bytes
コンパイル時間 950 ms
コンパイル使用メモリ 101,872 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-12-23 23:06:59
合計ジャッジ時間 3,468 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 99
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘void solve()’:
main.cpp:105:18: warning: format ‘%d’ expects argument of type ‘int’, but argument 2 has type ‘ULL’ {aka ‘long long unsigned int’} [-Wformat=]
  105 |         printf("%d\n", nCk(aki+kind,kind) );
      |                 ~^     ~~~~~~~~~~~~~~~~~~
      |                  |        |
      |                  int      ULL {aka long long unsigned int}
      |                 %lld
main.cpp:87:14: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   87 |         scanf("%d",&N);
      |         ~~~~~^~~~~~~~~

ソースコード

diff # 

#ifdef __GNUC__
#pragma GCC optimize ("O3")
#pragma GCC target ("avx")
#endif

#define _USE_MATH_DEFINES

#include <iostream>
#include <iomanip>
#include <stdio.h>

#include <sstream>
#include <algorithm>
#include <cmath>

#include <string>
#include <cstring>
#include <vector>
#include <valarray>
#include <array>

#include <queue>
#include <complex>
#include <set>
#include <map>
#include <stack>
#include <list>

#include <cassert>//assert();
#include <fstream>
/////////
#define REP(i, x, n) for(int i = x; i < n; i++)
#define rep(i,n) REP(i,0,n)

/////////
typedef long long LL;
typedef long double LD;
typedef unsigned long long ULL;
#define PII pair<int,int>
/////////
using namespace::std;

// 最大公約数
template<class T>
inline T gcd(T a, T b){return b == 0 ? a : gcd(b, a % b);}
// 最小公倍数
template<class T>
inline T lcm(T a, T b){return a * b / gcd(a, b);}
////////////////////////////////
const ULL mod = 1000000007;
ULL inv[1000010];
ULL invMod(ULL k){//k**(mod-2)
//  098765432109876543210987654321
//0b111011100110101100101000000111
	ULL p = mod-2;
	ULL ret = 1;
	while(p){//30回回る
		if( p & 1 ){
			ret = (ret*k)%mod;
		}
		k = (k*k)%mod;
		p >>= 1;
	}
	return ret;
}

ULL nCk(ULL N,ULL k){
	k = min(k,N-k);
	if( k == 0 ){return 1;}
	if( k == 1 ){return N%mod;}
	ULL ret = 1;
	for(int i=1;i<=k;++i){
		ret *= (N+1-i);//OVF
		ret %= mod;
	}
	ULL kkai = 1;
	for(int i=1;i<=k;++i){
		kkai *= i;
		kkai %= mod;
	}
	ret = ( ret*invMod(kkai) ) %mod;
	return ret;
}

inline void solve(){
	int N;
	scanf("%d",&N);
	int temp	= 0;
	int sum		= 0;
	int kind	= 0;

	while( scanf("%d",&temp) != EOF ){
		sum += temp;
		++kind;
	}
	if( sum == 0 ){
		printf("1\n");
		return;
	}
	if( sum + kind-1 > N ){
		printf("NA\n");
		return;
	}
	int aki = N - (sum+kind-1);
	printf("%d\n", nCk(aki+kind,kind) );
}

int main(void){
	//std::cin.tie(0);
	//std::ios::sync_with_stdio(false);
	//std::cout << std::fixed;//小数を10進数表示
	//cout << setprecision(16);//小数をいっぱい表示する。16?

	solve();

	return 0;
}
0