Submission #1191592

#TimeUsernameProblemLanguageResultExecution timeMemory
119159212345678Flooding Wall (BOI24_wall)C++20
36 / 100
547 ms315292 KiB
#include <bits/stdc++.h>

using namespace std;

#define ll long long

const ll nx=1e4+5, kx=1e3+5, mod=1e9+7;

ll n, a[nx], b[nx], dpl[nx][kx], dpr[nx][kx], ans, qsl[nx][kx], qsr[nx][kx];

int main()
{
    cin.tie(NULL)->sync_with_stdio(false);
    cin>>n;
    for (int i=1; i<=n; i++) cin>>a[i];
    for (int i=1; i<=n; i++) cin>>b[i];
    dpl[0][0]=1;
    for (int i=1; i<=n; i++)
    {
        for (int j=b[i]; j<kx; j++) dpl[i][j]=(dpl[i][j]+dpl[i-1][j])%mod;
        for (int j=a[i]; j<kx; j++) dpl[i][j]=(dpl[i][j]+dpl[i-1][j])%mod;
        for (int j=0; j<b[i]; j++) dpl[i][b[i]]=(dpl[i][b[i]]+dpl[i-1][j])%mod;
        for (int j=0; j<a[i]; j++) dpl[i][a[i]]=(dpl[i][a[i]]+dpl[i-1][j])%mod;
        for (int j=kx-2; j>=0; j--) qsl[i][j]=(qsl[i][j+1]+dpl[i][j])%mod;
        //cout<<"debug "<<i<<' '<<dpl[i][1]<<' '<<dpl[i][2]<<'\n';
    }
    dpr[n+1][0]=1;
    for (int i=n; i>=1; i--)
    {
        for (int j=b[i]; j<kx; j++) dpr[i][j]=(dpr[i][j]+dpr[i+1][j])%mod;
        for (int j=a[i]; j<kx; j++) dpr[i][j]=(dpr[i][j]+dpr[i+1][j])%mod;
        for (int j=0; j<b[i]; j++) dpr[i][b[i]]=(dpr[i][b[i]]+dpr[i+1][j])%mod;
        for (int j=0; j<a[i]; j++) dpr[i][a[i]]=(dpr[i][a[i]]+dpr[i+1][j])%mod;
        for (int j=kx-2; j>=0; j--) qsr[i][j]=(qsr[i][j+1]+dpr[i][j])%mod;
    }
    for (int i=1; i<=n; i++)
    {
        for (int j=1; j+a[i]<kx-1; j++) 
        {
            ans=(ans+(dpl[i-1][j+a[i]]*qsr[i+1][j+a[i]+1])%mod*j)%mod;
            ans=(ans+(qsl[i-1][j+a[i]+1]*dpr[i+1][j+a[i]])%mod*j)%mod;
            ans=(ans+(dpl[i-1][j+a[i]]*dpr[i+1][j+a[i]])%mod*j)%mod;
        }
        for (int j=1; j+b[i]<kx-1; j++)
        {
            ans=(ans+(dpl[i-1][j+b[i]]*qsr[i+1][j+b[i]+1])%mod*j)%mod;
            ans=(ans+(qsl[i-1][j+b[i]+1]*dpr[i+1][j+b[i]])%mod*j)%mod;
            ans=(ans+(dpl[i-1][j+b[i]]*dpr[i+1][j+b[i]])%mod*j)%mod;
        }
    }
    cout<<ans;
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...