ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

ABC222 D - Between Two Arrays(dp,差分优化)

2021-10-15 15:35:16  阅读:171  来源: 互联网

标签:temp Arrays maxm Two cin int Between define mod


题意:

在这里插入图片描述

解法:

d[i][j]表示前i个数,c[]末尾为j的方案数.
设l=a[i],r=b[i].
发现每次转移需要将d[i][j]加到d[i+1][]的一个区间,
那么复杂度就变成O(n^3)的了.

但是由于是连续区间,所以可以用差分优化以下,
复杂度就降为O(n^2)了

code:

#include<bits/stdc++.h>
// #define MULTI_CASE
#define SYNC_OFF
#define PI pair<int,int>
#define ll long long
#define int long long
using namespace std;
const int mod=998244353;
// const int mod=1e9+7;
const int maxm=3e3+5;
int d[maxm][maxm];
int temp[maxm];
int a[maxm];
int b[maxm];
int n;
void solve(){
    cin>>n;
    for(int i=1;i<=n;i++){
        cin>>a[i];
    }
    for(int i=1;i<=n;i++){
        cin>>b[i];
    }
    //dp[i][j]表示前i个数,末尾为j的方案数
    d[0][0]=1;
    for(int i=0;i<n;i++){
        for(int j=0;j<=3e3;j++){
            temp[j]=0;
        }
        int l=a[i+1],r=b[i+1];
        for(int j=0;j<=3e3;j++){
            if(!d[i][j])continue;
            if(j<l){
                temp[l]=(temp[l]+d[i][j])%mod;
                temp[r+1]=(temp[r+1]-d[i][j])%mod;
            }else if(j<=r){
                temp[j]=(temp[j]+d[i][j])%mod;
                temp[r+1]=(temp[r+1]-d[i][j])%mod;
            }
        }
        for(int j=0;j<=3e3;j++){
            temp[j+1]=(temp[j+1]+temp[j])%mod;
            d[i+1][j]=temp[j];
        }
    }
    int ans=0;
    for(int i=0;i<=3e3;i++){
        ans=(ans+d[n][i])%mod;
    }
    ans=(ans%mod+mod)%mod;
    cout<<ans<<endl;
}
void Main(){
    #ifdef MULTI_CASE
    int T;cin>>T;while(T--)
    #endif
    solve();
}
void Init(){
    #ifdef SYNC_OFF
    ios::sync_with_stdio(0);cin.tie(0);
    #endif
    #ifndef ONLINE_JUDGE
    freopen("../in.txt","r",stdin);
    freopen("../out.txt","w",stdout);
    #endif
}
signed main(){
    Init();
    Main();
    return 0;
}

标签:temp,Arrays,maxm,Two,cin,int,Between,define,mod
来源: https://blog.csdn.net/weixin_44178736/article/details/120784204

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有