제출 #852094

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
852094		vjudge1	Calvinball championship (CEOI15_teams)	C++17	100 / 100	342 ms	952 KiB

teams

#include<bits/stdc++.h>
#define pb push_back
#define pli pair<int,int>
#define fi first
#define se second
#define fastio ios_base::sync_with_stdio(false); cin.tie(NULL);
using namespace std;
using ll=long long;
const ll maxN=1e4+10;
const ll inf=1e18;
const ll mod=1e6+7;
void add(ll &x,ll y)
{
    x+=y;
    if(x>=mod) x-=mod;
}
ll a[maxN],dp[2][maxN][2],n;
void solve()
{
    cin >> n;
    for(int i=1;i<=n;i++)
    {
        cin >> a[i];
    }
    dp[0][0][0]=1;
    for(int i=0;i<n;i++)
    {
        for(int j=1;j<=i+1;j++)
        {
            dp[(i+1)&1][j][0]=dp[(i+1)&1][j][1]=0;
        }
        ll st;
        if(i==0) st=0;
        else st=1;
        for(int j=st;j<=i;j++)
        {
            if(a[i+1]<=j+1)
            {
                ll x;
                if(a[i+1]==j+1) x=j+1;
                else x=j;
                add(dp[(i+1)&1][x][0],dp[i&1][j][0]);
            }
            ll cc=min(a[i+1]-1,1ll*j);
            add(dp[(i+1)&1][j][1],cc*dp[i&1][j][0]%mod);
            if(j+1<a[i+1])
            {
                add(dp[(i+1)&1][j+1][1],dp[i&1][j][0]);
            }
            add(dp[(i+1)&1][j][1],j*dp[i&1][j][1]%mod);
            add(dp[(i+1)&1][j+1][1],dp[i&1][j][1]);
        }
    }
    ll ans=0;
    for(int i=1;i<=n;i++)
    {
        add(ans,dp[n&1][i][0]);
        add(ans,dp[n&1][i][1]);
    }
    cout << ans;
}
int main()
{
    fastio
    //freopen(TASKNAME".INP","r",stdin);
    //freopen(TASKNAME".OUT","w",stdout);
    solve();
}

• Subtask 1: 10 / 10
• Subtask 2: 10 / 10
• Subtask 3: 10 / 10
• Subtask 4: 10 / 10
• Subtask 5: 10 / 10
• Subtask 6: 10 / 10
• Subtask 7: 10 / 10
• Subtask 8: 10 / 10
• Subtask 9: 10 / 10
• Subtask 10: 10 / 10