2487 小b和环
- 2 秒
- 262,144 KB
- 20 分
- 3 级题
小b有一个长度为n的环,每个点上有个数字。
现在请你选出一些点,满足选出的任意两个点在环上不相邻,且选出的点的数字之和最大,你只需输出这个最大值。
收起
输入
第一行输入一个数n,其中0<n≤50000; 第二行输入n个非负整数,第i个数表示环上顺时针第i个点上的数字,以空格隔开。 0<=每个点上的数字<=10000。
输出
输出一个数,表示最大值。
输入样例
4 1 2 3 1
输出样例
4
题解:刚开始我是这么想的,dp[i] = max(dp[i-2] , dp[i-3]) + a[i] ,前 i 位 的值由前 i-2 或 前 i-3 递推而来,最基本的递推方程已经确立了;随后交了一发,只过了一半。
wa :一维的dp只能维护 1 - n 之间的不是相邻的,不能保证 1 和 n 是否相邻。
随后我又想了想,选了n就不能选1,只能再开一维数组标记前边是否将起始点 1 包含了进去。
#include<set>
#include<map>
#include<list>
#include<queue>
#include<stack>
#include<math.h>
#include<vector>
#include<bitset>
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<iostream>
#include<algorithm>
#define eps (1e-8)
#define MAX 0x3f3f3f3f
#define u_max 1844674407370955161
#define l_max 9223372036854775807
#define i_max 2147483647
#define re register
#define nth(k,n) nth_element(a,a+k,a+n); // 将 第K大的放在k位
#define ko() for(int i=2;i<=n;i++) s=(s+k)%i // 约瑟夫
#define ok() v.erase(unique(v.begin(),v.end()),v.end()) // 排序,离散化
using namespace std;
inline int read(){
char c = getchar(); int x = 0, f = 1;
while(c < '0' || c > '9') {if(c == '-') f = -1; c = getchar();}
while(c >= '0' & c <= '9') x = x * 10 + c - '0', c = getchar();
return x * f;
}
typedef long long ll;
const double pi = atan(1.)*4.;
const int M=1e3+5;
const int N=5e4+5;
// dp[i][0] 前 i 位不包含第一位的最大值
// dp[i][1] 前 i 位包含第一位的最大值
ll dp[N][2],a[N];
int main(){
int n;
scanf("%d",&n);
for(int i=1;i<=n;i++)
scanf("%lld",&a[i]);
if(n<4){
if(n==1) printf("%lld\n",a[1]);
else if(n==2) printf("%lld\n",max(a[1],a[2]));
else printf("%lld\n",max(a[3],max(a[1],a[2])));
return 0;
}
memset(dp,0,sizeof(dp));
dp[0][0]=0; dp[1][1]=a[1];
dp[2][0]=a[2]; dp[3][1]=dp[1][1]+a[3]; dp[3][0]=a[3];
for(int i=4;i<n;i++){
dp[i][1]=max(dp[i-2][1],dp[i-3][1])+a[i];
dp[i][0]=max(dp[i-2][0],dp[i-3][0])+a[i];
}
dp[n][0]=max(dp[n-2][0],dp[n-3][0])+a[n]; // 第 n 位只能取 0(0 代表不包含第一位
ll ans=0;
for(int i=1;i<=n;i++){
ans=max(ans,dp[i][1]);
ans=max(ans,dp[i][0]);
}
printf("%lld\n",ans);
return 0;
}