AGC 030B.Tree Burning(贪心)

Posted 2021-02-07 sovietpower

题目链接

---

(Description)

有一个长为(L)的环，上面有(n)棵树，坐标分别为(a_i)。初始时在原点。
每次你可以选择顺时针或逆时针走到第一棵没有被烧掉的树，停在这个位置，然后烧掉这棵树。重复这一过程直到所有树都被烧掉。
求走的总路程最多可以是多少。
(nleq2 imes10^5, a_i,Lleq10^9)。

(Solution)

真的菜啊QAQ 当时连这个都不会

记顺时针走一次为(L)，逆时针走一次为(R)。
初步想法是(LRLRLR...)这样走。但显然有反例（比如两棵树分别在(1, L-1)）。
事实上用(LLLLRLR...)和(RRRLRLR...)，这两种走法就没问题了。（具体证明没有大概既然开始来回走了，就比继续往一个方向走更优吧？）
前缀和后缀和预处理一下，枚举刚开始反复横跳进行(LRLR...)的位置就可以了。

---

//8ms   4608KB
#include <cstdio>
#include <cctype>
#include <algorithm>
//#define gc() getchar()
#define MAXIN 500000
#define gc() (SS==TT&&(TT=(SS=IN)+fread(IN,1,MAXIN,stdin),SS==TT)?EOF:*SS++)
typedef long long LL;
const int N=2e5+5;

int A[N];
char IN[MAXIN],*SS=IN,*TT=IN;

inline int read()
{
    int now=0;register char c=gc();
    for(;!isdigit(c);c=gc());
    for(;isdigit(c);now=now*10+c-48,c=gc());
    return now;
}
LL Solve(const int n,const int L)
{
    static LL pre[N],suf[N];
    for(int i=1; i<=n; ++i) pre[i]=pre[i-1]+A[i];
    for(int i=n; i; --i) suf[i]=suf[i+1]+L-A[i];
    LL ans=A[n];
    for(int i=1,p; i<n; ++i)
        p=i+((n-i)>>1), (n-i)&1 ? ans=std::max(ans,(suf[p+2]+pre[p]-pre[i]+A[i])*2+L-A[p+1]) : ans=std::max(ans,(suf[p+1]+pre[p-1]-pre[i]+A[i])*2+A[p]);
    return ans;
}

int main()
{
    const int L=read(),n=read();
    for(int i=1; i<=n; ++i) A[i]=read();
    LL ans=Solve(n,L);
    std::reverse(A+1,A+1+n);
    for(int i=1; i<=n; ++i) A[i]=L-A[i];
    ans=std::max(ans,Solve(n,L));
    printf("%lld
",ans);

    return 0;
}