結果

問題 No.1694 ZerOne
コンテスト
ユーザー planes
提出日時 2021-10-02 03:25:24
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 116 ms / 2,000 ms
コード長 972 bytes
コンパイル時間 1,540 ms
コンパイル使用メモリ 176,192 KB
実行使用メモリ 109,696 KB
最終ジャッジ日時 2024-07-19 20:54:48
合計ジャッジ時間 3,347 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 31
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include <bits/stdc++.h> 
using namespace std;
using ll =long long;
#define all(v) v.begin(),v.end()
 #define rep(i,a,b) for(int i=a;i<b;i++)
#define rrep(i,a,b) for(int i=a;i>=b;i--)

vector<vector<vector<ll>>> dp;
vector<ll> A;
 string S;

ll solve(ll i,ll k,ll p) {

    if(i==0) {
        if(k==0&&p==0) return 1;
        else return 0;
    }
        
    if(dp[i][k][p+1800]!=-1) {
        return dp[i][k][p+1800];
    }

    ll s=0;
    if(k==0) {
        s+=solve(i-1,k,p);
    }
    else{

    for(ll j=i-1;j>=0;j--) {
        s+=solve(j,k-1,A[k-1]-j+p);
    }
    }

    dp[i][k][p+1800]=s;
    return s;

}


int main() {
   cin>>S;
ll N=0;
for(ll i=0;i<S.size();i++) {
    if(S[i]=='1') N++;
}

A=vector<ll> (N);
ll now=0;
for(ll i=0;i<S.size();i++) {
    if(S[i]=='1') {
        A[now]=i;
        now++;
    }
}
/**/
dp=vector<vector<vector<ll>>> (S.size()+1,vector<vector<ll>> (N+1,vector<ll> (3600,-1)));
dp[0][0][0]=1;

cout<<solve(S.size(),N,0LL)<<endl;
}
0