oj.uz

Submission #852094

# Submission time Handle Problem Language Result Execution time Memory
852094 2023-09-21T08:24:55 Z vjudge1 Calvinball championship (CEOI15_teams) C++17
100 / 100
 342 ms 952 KB 
#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.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 344 KB Output is correct
2 Correct 1 ms 344 KB Output is correct
3 Correct 0 ms 344 KB Output is correct
4 Correct 0 ms 344 KB Output is correct
5 Correct 1 ms 344 KB Output is correct
6 Correct 0 ms 344 KB Output is correct

Subtask #2
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 344 KB Output is correct
2 Correct 0 ms 344 KB Output is correct
3 Correct 1 ms 344 KB Output is correct

Subtask #3
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 344 KB Output is correct
2 Correct 0 ms 344 KB Output is correct
3 Correct 0 ms 348 KB Output is correct

Subtask #4
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 344 KB Output is correct

Subtask #5
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 344 KB Output is correct
2 Correct 0 ms 344 KB Output is correct
3 Correct 1 ms 344 KB Output is correct

Subtask #6
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 344 KB Output is correct
2 Correct 1 ms 344 KB Output is correct
3 Correct 1 ms 348 KB Output is correct

Subtask #7
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 4 ms 344 KB Output is correct
2 Correct 4 ms 344 KB Output is correct
3 Correct 4 ms 344 KB Output is correct

Subtask #8
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 311 ms 952 KB Output is correct

Subtask #9
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 89 ms 600 KB Output is correct
2 Correct 85 ms 600 KB Output is correct
3 Correct 88 ms 852 KB Output is correct

Subtask #10
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 342 ms 860 KB Output is correct
2 Correct 325 ms 848 KB Output is correct
3 Correct 329 ms 932 KB Output is correct