标签:
输入一个含有 n(1≤n≤100000) 个非负整数的 a 数组和一个 1~n 的排列 p 数组,求每次删除 a[p[i]] 后,最大连续子段和(不能跨越被删除的)是多少?
因为都是非负整数,答案一定是尽量长的区间和。
s[i] 表示 a 的前缀和,区间(l,r]的和就是s[r]-s[l]。
我们用 STL 里的 set 存区间,一开始只有(0,n]区间,删去第一个数后,就要去掉(0,n]区间,加上(0,p[1]-1]和(p[1],n]区间,类似地每次删除一个数,就要去掉包含它的区间,加上两个新区间,同时用 multiset 储存下区间和,每次输出最大的区间和,删除时也删除掉对应的区间和。
需要注意的细节:
URL:http://codeforces.com/contest/722/problem/C
代码:
#include<bits/stdc++.h> #define ll long long #define N 100005 using namespace std; int n; ll s[N]; set< pair<int,int> > seg; multiset<ll> sum; void erase(int p){ set< pair<int,int> > ::iterator it=seg.lower_bound(make_pair(p,0)); sum.erase(sum.find(s[it->first]-s[it->second])); seg.insert(make_pair(p-1,it->second)),sum.insert(s[p-1]-s[it->second]); seg.insert(make_pair(it->first,p)),sum.insert(s[it->first]-s[p]); seg.erase(it); } ll max(){ multiset<ll> ::reverse_iterator mit=sum.rbegin(); return *mit; } int main(){ scanf("%d",&n); for(int i=1;i<=n;i++){ int a; scanf("%d",&a); s[i]=s[i-1]+a; } seg.insert(make_pair(n,0)); sum.insert(s[n]); for(int i=1;i<=n;i++){ int p; scanf("%d",&p); erase(p); printf("%I64d\n",max()); } }
看到一个不用set且更快的代码,大概思路是,倒过来依次放上删除的数,然后找最大连续和。
#include<bits/stdc++.h> using namespace std; int x[100002],y[100002],Seg[100002][2]; long long sum[100002],maxm; bool memo[100002]; vector<long long> ans; int main() { int n,i; scanf("%d",&n); for(i=1;i<=n;i++) scanf("%d",&x[i]); for(i=1;i<=n;i++) scanf("%d",&y[i]); for(i=1;i<=n;i++) sum[i] = x[i] + sum[i-1]; for(i=n;i>0;i--) { // y[i] == ME ( element to be built ) Seg[y[i]][0] = y[i]; Seg[y[i]][1] = y[i]; memo[y[i]] = 1; if(memo[y[i]-1]) Seg[y[i]][0] = Seg[y[i]-1][0]; if(memo[y[i]+1]) Seg[y[i]][1] = Seg[y[i]+1][1]; Seg[Seg[y[i]][0]][1] = Seg[y[i]][1]; Seg[Seg[y[i]][1]][0] = Seg[y[i]][0]; maxm = max(maxm,sum[Seg[y[i]][1]] - sum[Seg[y[i]][0]-1]); ans.push_back(maxm); } for(i=ans.size()-2;i>=0;i--) printf("%I64d\n",ans[i]); printf("0\n"); }
【Codeforces 722C】Destroying Array (数据结构、set)
标签:
原文地址:http://www.cnblogs.com/flipped/p/5928466.html