結果

問題 No.117 組み合わせの数
ユーザー pppppp
提出日時 2020-04-21 16:58:24
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
RE  
実行時間 -
コード長 5,002 bytes
コンパイル時間 950 ms
コンパイル使用メモリ 102,612 KB
実行使用メモリ 27,012 KB
最終ジャッジ日時 2024-04-17 05:26:47
合計ジャッジ時間 2,542 ms
ジャッジサーバーID
(参考情報)
judge2 / judge1
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 RE -
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function 'void init()':
main.cpp:160:15: warning: iteration 999999 invokes undefined behavior [-Waggressive-loop-optimizations]
  160 |         inv[i]=(mod-mod/i)*inv[mod%i];
      |         ~~~~~~^~~~~~~~~~~~~~~~~~~~~~~
main.cpp:157:18: note: within this loop
  157 |     for (ll i=2;i<=1000001;++i){
      |                 ~^~~~~~~~~
main.cpp:166:16: warning: iteration 1000000 invokes undefined behavior [-Waggressive-loop-optimizations]
  166 |         fact[i]=fact[i-1]*i;
      |         ~~~~~~~^~~~~~~~~~~~
main.cpp:165:18: note: within this loop
  165 |     for (ll i=1;i<=1000001;++i){
      |                 ~^~~~~~~~~

ソースコード

diff #

#include <fstream>
#include <iostream>
#include <algorithm>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <string>
#include <sstream>
#include <map>
#include <set>
#include <vector>
#include <stack>
#include <cmath>
#include <queue>
#include <random>
#include <numeric>

using namespace std;

#define INT_MAX_VALUE 2147483647
#define LONG_LONG_MAX_VALUE 9223372036854775807
#define ll long long
#define ld long double

struct XX{
    ll d;
    int i;
    ll sita=0;
    ll z;
};

class xxGreater {
public:
    bool operator()(const XX& riLeft, const XX& riRight) const {
        //第2条件
        if((riLeft.z) == (riRight.z)){
            return riLeft.i > riRight.i;//<:昇順(小さいものから順番)、>:降順(大きいものから順番)
            //プライオリティキューの場合は > で、top()すると値の小さいものがとれる
        }
        //第1条件
        return (riLeft.z) > (riRight.z);
    }
};

class xx2Greater {
public:
    bool operator()(const XX& riLeft, const XX& riRight) const {
        return (riLeft.i) < (riRight.i);
    }
};


//map<long long,long long> prime_f(long long n){
//    map<long long,long long>res;
//    for(int i=2;i*i<=n;i++){
//        while(n%i==0){
//            ++res[i];
//            n/=i;
//        }
//    }
//    if(n!=1)res[n]=1;
//    return res;
//}

//int n;
////int dat[2*10000000];
////int dat2[2*10000000];
//int dat[10];
//int dat2[10];
//
//void init(int n_){
//    n=1;
//    while(n<n_)n*=2;
//    for(int i=0;i<2*n-1;i++){
//        dat[i]=0;
//        dat2[i]=0;
//    }
//}
//
//void initset(int k,int a){
//    k+=n-1;
//    dat[k]=a;
//    while(k>0){
//        k=(k-1)/2;
//        dat[k]=dat[k*2+1]+dat[k*2+2];
//    }
//}
//
////[a,b)の間を[l,r]区間で比較しアップデート
////引数のindexに注意
////nは固定。initで計算すみ
////update2(L[i],R[i]+1,0,0,n,D[i]);
//void update2(int a,int b,int k,int l,int r,int v){//v更新値、区間は0-index
//    if(r<=a || b<=l)return;
//    if(a<=l && r<=b){
//        dat[k]+=dat2[k];
//        if(r-l>1){
//            dat2[k*2+1]+=dat2[k]/2;
//            dat2[k*2+1]+=dat2[k]/2;
//        }
//        dat2[k]=v*(r-l);
//        return;
//    }else{
//        update2(a,b,k*2+1,l,(l+r)/2,v);
//        update2(a,b,k*2+2,(l+r)/2,r,v);
//        return;
//    }
//}
//
//int query(int a,int b,int k,int l,int r){
//    if(r<=a || b<=l)return 0;
//    if(a<=l && r<=b){
//        dat[k]+=dat2[k];
//        if(r-l>1){
//            dat2[k*2+1]+=dat2[k]/2;
//            dat2[k*2+1]+=dat2[k]/2;
//        }
//        dat2[k]=0;
//        return dat[k];
//    }
//    else{
//        int vl=query(a,b,k*2+1,l,(l+r)/2);
//        int vr=query(a,b,k*2+2,(l+r)/2,r);
//        return vl+vr;
//    }
//}



//void printb(unsigned int v) {
//    unsigned int mask = (int)1 << (sizeof(v) * CHAR_BIT - 1);
//    do putchar(mask & v ? '1' : '0');
//    while (mask >>= 1);
//}




#ifdef DEBUG
int a[200];
ll fact[1000001];
ll invfact[1000001];
ll inv[1000001];
#else
int a[200000];
ll fact[1000001];
ll invfact[1000001];
ll inv[1000001];
#endif

ll mod=1000000007;


void init() {
    inv[0]=0;
    inv[1]=1;
    for (ll i=2;i<=1000001;++i){
        //p=(p/a)*a+(p%a)はmod pで0なので(p/a)*a=-(p%a)
        //よってaの逆元は、-(p/a)*(p%a)^(-1)
        inv[i]=(mod-mod/i)*inv[mod%i];
        inv[i]%=mod;
    }
    fact[0]=1;
    invfact[0]=1;
    for (ll i=1;i<=1000001;++i){
        fact[i]=fact[i-1]*i;
        fact[i]%=mod;
        invfact[i]=invfact[i-1]*inv[i];
        invfact[i]%=mod;
    }
}
ll cmb(ll n, ll r) {
    if(r<0 || r>n) return 0;
    return fact[n]%mod*invfact[r]%mod*invfact[n-r]%mod;
}
ll prm(ll n, ll r) {
    if(r<0 || r>n) return 0;
    return fact[n]*invfact[n-r]%mod;
}
ll hcmb(int n,int r) {
    if(n==0 && r==0) return 1;
    return cmb(n+r-1,r);
}
int main(int argc, const char * argv[])
{
    //scanf("%s",S);
    //scanf("%d",&N);
    //scanf("%lld %lld",&target1,&target2);
    //sscanf(tmp.c_str(),"%dd%d%d",&time[i], &dice[i], &z[i]);
    //getline(cin, target);
    //ifstream ifs("a.txt");//テスト用
    //ifs >> a;
    //ここから
    
    //入力高速化
    ios::sync_with_stdio(false);
    cin.tie(0);
    ifstream ifs("in.txt");//テスト用

    int n;
    cin>>n;
    init();
    for(int i=0;i<n;i++){
        string tmp;
        cin>>tmp;
        int n=0;
        int r=0;
        int ind=2;
        while(tmp[ind]!=','){
            n=n*10+tmp[ind]-'0';
            ind++;
        }
        ind++;
        while(tmp[ind]!=')'){
            r=r*10+tmp[ind]-'0';
            ind++;
        }
        if(tmp[0]=='C'){
            cout << cmb(n,r) << endl;
        }else if(tmp[0]=='P'){
            cout << prm(n,r) << endl;
        }else{
            cout << hcmb(n,r) << endl;
        }
    }
    
    //ここまで
    //cout << "ans" << endl;
    //printf("%.0f\n",ans);//小数点以下表示なし
    //printf("%.7f\n",p);
    
    return 0;
}
0