제출 #1038164

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
1038164		Mr_Ph	Flooding Wall (BOI24_wall)	C++17	56 / 100	383 ms	1048576 KiB

Main

#include<bits/stdc++.h>
#define int long long
using namespace std;
const int mod=(int)1e9+7;
vector<vector<int>>dpl;
vector<vector<int>>dpr;
int getsuml(int i,int l,int r)
{
    return (dpl[i][r]-dpl[i][l-1]+mod)%mod;
}
int getsumr(int i,int l,int r)
{
    return (dpr[i][r]-dpr[i][l-1]+mod)%mod;
}
signed main()
{
    int n;
    cin>>n;
    vector<int>arr(n+1);
    vector<int>brr(n+1);
    for(int i=1;i<=n;i++)cin>>arr[i];
    for(int i=1;i<=n;i++)cin>>brr[i];
    set<int>st;
    for(auto i:arr)st.insert(i);
    for(auto i:brr)st.insert(i);
    int cnt=1;
    vector<int>hi;
    hi.push_back(0);
    for(auto i:st)
    {
        cnt++;
        hi.push_back(i);
    }
    int a7a=*max_element(arr.begin(),arr.end());
    a7a=max(a7a,*max_element(brr.begin(),brr.end()));
    a7a=lower_bound(hi.begin(),hi.end(),a7a)-hi.begin();
    dpl.resize(n+3);
    dpr.resize(n+3);
    for(int i=0;i<=n+1;i++)
    {
        dpl[i].resize(a7a+3);
        dpr[i].resize(a7a+3);
    }
    int ans=0;
    // cout<<dpl.size();
    // return 0;
    dpl[0][0]=1;
    for(int i=1;i<=n;i++)arr[i]=lower_bound(hi.begin(),hi.end(),arr[i])-hi.begin(),brr[i]=lower_bound(hi.begin(),hi.end(),brr[i])-hi.begin();
    for(int i=1;i<=n;i++)
    {
        for(int j=arr[i];j>=0;j--)
        {
            if(dpl[i-1][j]==0)continue;
            dpl[i][arr[i]]+=dpl[i-1][j];
            dpl[i][arr[i]]%=mod;
        }
        for(int j=arr[i]+1;j<=a7a;j++)
        {
            if(dpl[i-1][j]==0)continue;
            dpl[i][j]+=dpl[i-1][j];
            dpl[i][j]%=mod;
        }
        for(int j=brr[i];j>=0;j--)
        {
            if(dpl[i-1][j]==0)continue;
            dpl[i][brr[i]]+=dpl[i-1][j];
            dpl[i][brr[i]]%=mod;
        }
        for(int j=brr[i]+1;j<=a7a;j++)
        {
            if(dpl[i-1][j]==0)continue;
            dpl[i][j]+=dpl[i-1][j];
            dpl[i][j]%=mod;
        }
    }
    dpr[n+1][0]=1;
    for(int i=n;i>=1;i--)
    {
        for(int j=arr[i];j>=0;j--)
        {
            if(dpr[i+1][j]==0)continue;
            dpr[i][arr[i]]+=dpr[i+1][j];
            dpr[i][arr[i]]%=mod;
        }
        for(int j=arr[i]+1;j<=a7a;j++)
        {
            if(dpr[i+1][j]==0)continue;
            dpr[i][j]+=dpr[i+1][j];
            dpr[i][j]%=mod;
        }
        for(int j=brr[i];j>=0;j--)
        {
            if(dpr[i+1][j]==0)continue;
            dpr[i][brr[i]]+=dpr[i+1][j];
            dpr[i][brr[i]]%=mod;
        }
        for(int j=brr[i]+1;j<=a7a;j++)
        {
            if(dpr[i+1][j]==0)continue;
            dpr[i][j]+=dpr[i+1][j];
            dpr[i][j]%=mod;
        }
    }
    // cout<<ans<<endl;
    for(int i=0;i<=n+1;i++)
        for(int j=1;j<=a7a;j++)dpl[i][j]+=dpl[i][j-1],dpl[i][j]%=mod,dpr[i][j]+=dpr[i][j-1],dpr[i][j]%=mod;
    for(int i=1;i<=n;i++)
    {
        for(int j=arr[i];j<=a7a;j++)
        {
            int x=getsuml(i-1,j,j);
            x*=getsumr(i+1,j,a7a);
            x%=mod;
            ans+=((hi[j]-hi[arr[i]])*x)%mod;
            x=getsumr(i+1,j,j);
            x*=getsuml(i-1,j+1,a7a);
            x%=mod;
            ans+=((hi[j]-hi[arr[i]])*x)%mod;
            ans%=mod;
        }
        for(int j=brr[i];j<=a7a;j++)
        {
            int x=getsuml(i-1,j,j);
            x*=getsumr(i+1,j,a7a);
            x%=mod;
            ans+=((hi[j]-hi[brr[i]])*x)%mod;
            x=getsumr(i+1,j,j);
            x*=getsuml(i-1,j+1,a7a);
            x%=mod;
            ans+=((hi[j]-hi[brr[i]])*x)%mod;
            ans%=mod;
        }
    }       
    cout<<ans<<endl;
}

• Subtask 1: 8 / 8
• Subtask 2: 17 / 17
• Subtask 3: 19 / 19
• Subtask 4: 0 / 14
• Subtask 5: 12 / 12
• Subtask 6: 0 / 30