代码根据此博客https://blog.csdn.net/acerandaker/article/details/82351251
输入样例1:
4
1 3 2 5
3 4 1 2 *
分析:逆着插入+并查集
#include<iostream>
#include<cmath>
#include<algorithm>
#include<string>
#include<string.h>
using namespace std;
#define lli long long int
const int num=100050;
int fa[num],vis[num],del[num];
lli aa[num];
lli sum[num];
int maxn=0;
int n,now;
int findd(int x)//找爹
{
if(fa[x]!=x)
return fa[x]=findd(fa[x]);
else
return x;
}
void mergee(int a,int b)//求和(左/右)
{
fa[b]=a;//爹发生变化
aa[a]+=aa[b];
return ;
}
int main()
{
cin>>n;lli mx=0;
for(int i=1; i<=n; i++)
{
cin>>aa[i];
fa[i]=i;
}
for(int i=1; i<=n; i++)
{
cin>>del[i];
}
for(int i=n; i>=1; i--)//考虑例1中*行,2、1、4、3
{
sum[i]=mx;
vis[del[i]]=1;
if(vis[del[i]-1]&&i>1)//(del[i]-1)>0,所以i>1。vis[ ]=1判断已经插入,插入后才能求和。
{
mergee(del[i],del[i]-1);//左和
mx=max(mx,aa[del[i]-1]);
}
if(vis[del[i]+1]&&i<n)//同上
{
mergee(findd(del[i]+1),del[i]); //右和
mx=max(mx,aa[fa[del[i]+1]]);
}
mx=max(mx,aa[del[i]]);
}
for(int i=1; i<=n; i++)
{
cout<<sum[i]<<endl;
}
return 0;
}
(正着分析在样例14超时了,所以也不知道对不对,希望大佬发现问题可以帮助解答,谢谢)
超时代码:
#include<iostream>
#include<cmath>
#include<string>
//#include<sring.h>
#include<algorithm>
#include<set>
using namespace std;
int n,m;
int aa[100001];
int bb[100001];
int book[10000001];
int maxn=0;
int main()
{
cin>>n;
for(int i=1; i<=n; i++)
{
cin>>aa[i];
}
for(int i=1; i<=n; i++)
{
cin>>bb[i];
}
int nowsum=0;
int maxsum=0;
for(int i=1; i<n; i++)
{
book[bb[i]]=1;
nowsum=0;
maxsum=0;
for(int j=1; j<=n; j++)
{
if(book[j]==1)
{
//aa[j]=0;
nowsum=0;
}
if(j!=bb[i])
{
nowsum+=aa[j];
if(nowsum>maxsum)
maxsum=nowsum;
}
else
{
aa[j]=0;
nowsum=0;
}
}
cout<<maxsum<<endl;
}
cout<<"0"<<endl;
}