題目連結

\(Description\)

給定\(n\)個數。你可以選擇一段區間將它們都加上或減去任意一個數。求最終序列中最多能有多少個數等於給定的\(C\)。
\(n\leq5\times10^5\)。

\(Solution\)

先記一個表示\(C\)的個數的字首和\(sum_i\)。
選擇修改的區間\([l,r]\)一定滿足\(A_l=A_r\)且都是由\(A_l\)變成\(C\)。所以我們列舉右端點，對每種權值單獨考慮。
那麼\(A_r\)要麼是由前面某個等於\(A_r\)的數轉移過來，要麼\(l\)直接等於\(r\)。
所以記\(mx_{a_i}\)為之前\(a_i\)這個數的最大貢獻，那麼\[mx_{a_i}=\max\{\ mx_{a_i}+1,\ \ sum_{i-1}+1\}\]

\(Ans=\max\{mx_{a_i}+sum_n-sum_i\}\)。
for一遍就行啦。

//31ms  6700KB
#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=5e5+5;

int A[N],sum[N],mx[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-'0',c=gc());
    return now;
}

int main()
{
    const int n=read(),C=read();
    for(int i=1; i<=n; ++i) sum[i]=sum[i-1]+((A[i]=read())==C);
    int ans=0; const int sn=sum[n];
    for(int i=1; i<=n; ++i)
        mx[A[i]]=std::max(mx[A[i]],sum[i-1])+1, ans=std::max(ans,mx[A[i]]+sn-sum[i]);
    printf("%d\n",ans);

    return 0;
}