题意:
给定长度为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;
}