[CSP-S 2024] 染色 AC满分代码

CSP-C++

已于 2024-11-01 11:19:23 修改

阅读量727

点赞数 8

CC 4.0 BY-SA版权

分类专栏： CSP c++ 文章标签：算法

于 2024-11-01 11:01:25 首次发布

本文链接：https://blog.youkuaiyun.com/qq_21167485/article/details/143426568

CSP c++ 专栏收录该内容

4 篇文章

订阅专栏

题目描述

给定一个长度为 nn 的正整数数组 AA，其中所有数从左至右排成一排。

你需要将 AA 中的每个数染成红色或蓝色之一，然后按如下方式计算最终得分：

设 CC 为长度为 nn 的整数数组，对于 AA 中的每个数 A_iAi（1 \leq i \leq n1≤i≤n）：

如果 A_iAi 左侧没有与其同色的数，则令 C_i = 0Ci=0。
否则，记其左侧与其最靠近的同色数为 A_jAj，若 A_i = A_jAi=Aj，则令 C_i = A_iCi=Ai，否则令 C_i = 0Ci=0。

你的最终得分为 CC 中所有整数的和，即 \sum \limits_{i=1}^n C_ii=1∑nCi。你需要最大化最终得分，请求出最终得分的最大值。

输入格式

本题有多组测试数据。

输入的第一行包含一个正整数 TT，表示数据组数。

接下来包含 TT 组数据，每组数据的格式如下：

第一行包含一个正整数 nn，表示数组长度。

第二行包含 nn 个正整数 A_1, A_2, \dots, A_nA1,A2,…,An，表示数组 AA 中的元素。

输出格式

对于每组数据：输出一行包含一个非负整数，表示最终得分的最大可能值。

输入输出样例

输入 #1复制

3
3
1 2 1
4
1 2 3 4
8
3 5 2 5 1 2 1 4

输出 #1复制

1
0
8

#include<bits/stdc++.h>
using namespace std;
long long a,p,x,n,t,f[1<<20];
main(){
    cin>>n;
    while(cin>>n){
        for(a=p=t=0,memset(f,-127,8<<20);n--;p=x){ 
            cin>>x;
            if(x == p){
                a+=x;
            }else{
                t=f[p]=max(f[p],max(f[x]+x,t));
            } 
        }
        cout<<a+t<<endl;
    }
}