1. 程式人生 > >Codeforces 332B Maximum Absurdity(DP+前綴和處理)

Codeforces 332B Maximum Absurdity(DP+前綴和處理)

http 枚舉 ostream ref eof 思路 AR int style

題目鏈接:http://codeforces.com/problemset/problem/332/B

題目大意:
給你n個數和一個整數k,要求找到不相交的兩個長度為k的區間,使得區間和最大,輸出這兩個區間的起點。
解題思路:
先計算前綴和,然後預處理出maxsum[i],maxsum[i]記錄i~n最大的長度為k子段的和。
然後再去枚舉即可。

 1 #include<cstdio>
 2 #include<iostream>
 3 #include<algorithm>
 4 #include<vector>
 5 #include<string
> 6 #include<string.h> 7 #include<cctype> 8 #include<math.h> 9 #include<stdlib.h> 10 #include<stack> 11 #include<queue> 12 #include<set> 13 #include<map> 14 #define lc(a) (a<<1) 15 #define rc(a) (a<<1|1) 16 #define MID(a,b) ((a+b)>>1) 17
#define fin(name) freopen(name,"r",stdin) 18 #define fout(name) freopen(name,"w",stdout) 19 #define clr(arr,val) memset(arr,val,sizeof(arr)) 20 #define _for(i,start,end) for(int i=start;i<=end;i++) 21 #define FAST_IO ios::sync_with_stdio(false);cin.tie(0); 22 using namespace std; 23 typedef long
long LL; 24 const int N=5e6+5; 25 const LL INF64=1e18; 26 const int INF=0x3f3f3f3f; 27 const double eps=1e-10; 28 29 LL sum[N]; //記錄前綴和 30 LL maxsum[N]; //maxsum[i]記錄i~n最大的長度為k子段的和 31 int idx[N]; //idx[i]記錄相應的i~n最大子段的起點 32 33 int main(){ 34 FAST_IO; 35 int n,k; 36 cin>>n>>k; 37 for(int i=1;i<=n;i++){ 38 cin>>sum[i]; 39 sum[i]+=sum[i-1]; 40 } 41 int lim=n-k+1; 42 for(int i=lim;i>=1;i--){ 43 LL now=sum[i+k-1]-sum[i-1]; //now為i~i+k-1之和 44 if(now>=maxsum[i+1]){ //因為要保證答案字典序最小,所以下標往小的取 45 maxsum[i]=now; 46 idx[i]=i; 47 } 48 else{ 49 maxsum[i]=maxsum[i+1]; 50 idx[i]=idx[i+1]; 51 } 52 } 53 54 int st1=1,st2=k+1; 55 LL mmax=sum[2*k]; 56 for(int i=1;i<=lim-k;i++){ 57 LL now=sum[i+k-1]-sum[i-1]; 58 if(now+maxsum[i+k]>mmax){ //maxsum[i+k]為i+k~n最大的長度為k的子段和 59 mmax=now+maxsum[i+k]; 60 st1=i,st2=idx[i+k]; 61 } 62 } 63 cout<<st1<<" "<<st2<<endl; 64 return 0; 65 }

Codeforces 332B Maximum Absurdity(DP+前綴和處理)