結果

問題 No.1685 One by One
ユーザー 宋来恒
提出日時 2023-01-08 16:15:27
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 255 ms / 3,000 ms
コード長 2,005 bytes
コンパイル時間 985 ms
コンパイル使用メモリ 71,896 KB
実行使用メモリ 503,580 KB
最終ジャッジ日時 2024-12-15 19:28:39
合計ジャッジ時間 7,158 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 46
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstring>
#include <vector>
using namespace std;
#define fi first
#define sc second
#define mkp make_pair
#define pii pair<int,int>
typedef long long ll;
const int N=8005,oo=1e9,mod=1e9+7;
inline int read() {
    int x=0,flag=0;char ch=getchar();
    while(ch<'0'||ch>'9') {flag|=(ch=='-');ch=getchar();}
    while('0'<=ch&&ch<='9') {x=(x<<3)+(x<<1)+ch-'0';ch=getchar();}
    return flag?-x:x;
}
inline int mx(int x,int y) {return x>y?x:y;}
inline int mn(int x,int y) {return x<y?x:y;}
inline void swp(int &x,int &y) {x^=y^=x^=y;}
inline int as(int x) {return x>0?x:-x;}

int n,m,C[N][N],S[N][N];
inline void prep() {
    C[0][0]=S[0][0]=1;
    C[1][0]=C[1][1]=1;S[1][0]=S[1][1]=1;
    for(int i=2;i<=m+n;++i) {
        C[i][0]=C[i][i]=1;
        for(int j=1;j<i;++j)
            C[i][j]=(C[i-1][j-1]+C[i-1][j])%mod;
        for(int j=0;j<=i;++j)
            S[i][j]=(S[i-2][j]+C[i][j])%mod;
    }
}
inline void solve_odd() {
    int ans=0,L=n/2;
    for(int i=0;i<=L;++i)
        for(int j=0;j<=L;++j) {
            if(i==0&&j==0) continue;
            ans=(ans+1ll*C[n][i]*C[n-i][j]%mod*S[m-1][i+j-1])%mod;
            if(i<=j) {
                if(m-n-i+j<0) continue;
                ans=(ans+1ll*C[n][i]*C[n-i][j]%mod*S[m-n-i+j+(i+j-1)][i+j-1])%mod;
            } else {
                if(m-n-j+i<0) continue;
                ans=(ans+1ll*C[n][i]*C[n-i][j]%mod*S[m-n-j+i+(i+j-1)][i+j-1])%mod;
            }
        }
    if(m%2==0||m>=n) ans=(ans+1)%mod;
    printf("%d\n",ans);
}
inline void solve_even() {
    int ans=0,L=n/2;
    for(int i=0;i<=L;++i)
        for(int j=0;j<L;++j) {
            if(i==0&&j==0) continue;
            ans=(ans+1ll*C[n][i]*C[n-i][j]%mod*S[m-1][i+j-1])%mod;
        }
    if(m%2==0) ans=(ans+1)%mod;
    printf("%d\n",ans);
}
int main() {
    n=read();m=read();
    if(m==0) {
        printf("1\n");
        return 0;
    }
    prep();
    if(n&1) solve_odd();
    else solve_even();
    return 0;
}
0