gym101194 C. Mr. Panda and Strips(set和multiset的使用,思维,预处理)

题意:

给定长度为n的序列a,
要求选出两个不相交的子区间,将他们拼接,
要求拼接出的序列所有数互不相同,
问拼接出的序列的最大长度是多少。

数据范围:1<=n<=1e3,1<=a[i]<=1e5

解法:
因为n<=1e3,因此最多只有1e3种颜色.

考虑固定一个区间[l,r],标记[l,r]中的所有颜色,
标记的颜色显然不能选,此时序列被切割为若干个子区间.

我们O(n^2)预处理d[l][r]表示[l,r]中满足颜色不重复的子区间最大长度.

那么可以用分割出的区间的max(d[][])+d[l][r]更新答案.

枚举固定的区间复杂度O(n^2),遍历切割出的区间O(n),总复杂度O(n^3).

枚举固定区间的复杂度是降不下去的,考虑将遍历切割出的区间这部分优化一下:
枚举固定区间时,[l,r]转移到[l,r+1],多了一个颜色a[r],
此时a[r]会将之前切割出的区间再切割为更小的区间,
遍历颜色为a[r]的所有位置p,
之前切割出的区间显然最多只有一个区间包含p,将这个区间从p位置分割为两部分即可.
因为位置只有n个,因此r从l一直右移的过程中,分割的总次数最多O(n),
考虑如何快速找到包含位置p的区间:
用set存区间(l,r),在set上二分即可.

如何快速计算当前所有区间的d[][]的最大值?
开一个multiset存d[][]即可.

算法总复杂度O(n^2 *log)
code:
#include<bits/stdc++.h>
#define PI pair<int,int>
using namespace std;
const int maxm=1e3+5;
vector<int>pos[maxm];
int d[maxm][maxm];
int mark[maxm];
int a[maxm];
int n;
void solve(){
    cin>>n;
    for(int i=1;i<=n;i++){
        cin>>a[i];
    }
    //将颜色离散化
    map<int,int>mp;
    int tot=0;
    for(int i=1;i<=n;i++){
        if(!mp[a[i]])mp[a[i]]=++tot;
        a[i]=mp[a[i]];
    }
    for(int i=1;i<=tot;i++)pos[i].clear();
    for(int i=1;i<=n;i++){
        pos[a[i]].push_back(i);
    }
    //预处理d[l][r]=[l,r]中满足条件的子区间最大长度.
    for(int i=1;i<=n;i++){
        for(int j=i;j<=n;j++){
            d[i][j]=0;
        }
    }
    for(int i=1;i<=n;i++){
        for(int j=1;j<=tot;j++)mark[j]=0;
        for(int j=i;j<=n;j++){
            if(mark[a[j]])break;
            mark[a[j]]=1;
            d[i][j]=j-i+1;
        }
    }
    for(int len=2;len<=n;len++){
        for(int i=1;i<=n;i++){
            int j=i+len-1;
            if(j>n)break;
            d[i][j]=max(d[i][j],max(d[i+1][j],d[i][j-1]));
        }
    }
    //
    int ans=0;
    for(int i=1;i<=n;i++){
        for(int j=1;j<=tot;j++)mark[j]=0;
        set<PI>se;//存区间,用来快速查找包含位置p的区间.
        multiset<int>s;//存每个区间的d[l][r]值,用于求max.
        se.insert({1,n});
        s.insert(d[1][n]);
        for(int j=i;j<=n;j++){
            if(mark[a[j]])break;
            mark[a[j]]=1;
            for(auto p:pos[a[j]]){
                auto it=se.upper_bound({p,n+1});
                it--;
                int l=it->first,r=it->second;
                se.erase(it);
                s.erase(s.find(d[l][r]));
                if(l<=p-1){
                    se.insert({l,p-1});
                    s.insert(d[l][p-1]);
                }
                if(p+1<=r){
                    se.insert({p+1,r});
                    s.insert(d[p+1][r]);
                }
            }
            if(!s.empty()){
                ans=max(ans,d[i][j]+*s.rbegin());
            }else{
                ans=max(ans,d[i][j]);
            }
        }
    }
    static int cas=1;
    cout<<"Case #"<<cas++<<": ";
    cout<<ans<<endl;
}
signed main(){
    ios::sync_with_stdio(0);
    int T;cin>>T;
    while(T--){
        solve();
    }
    return 0;
}

  • 2
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值