Pokémon Army (easy version)

题目链接

目录

• 题目分析
  • AC代码

题目分析

题意:

给你\(n\)个元素，你可以选其中\(k\)个元素构成一个子集\(b\)，子集的元素会以\(b_1-b_2+b_3-b_4\cdots\)的方式求和，问你怎样选让和最大

本题可以从dp的角度去分析，对于一个元素，我们有三种选择：不选，加上此元素，减去此元素。

这样本题就可以构成一个状态机dp，情况如下:

graph LR Enter("入口") Exit("出口") A("不选") B("加上") C("减去") Enter --> A --> C A --> B B --> C --> Exit C --> B --> Exit B --> A C --> A

1. 状态表示\(f(i,j)\):
   1. 集合：所有选前\(i\)个物品，且状态为\(j\)的集合
   2. 属性：Max
2. 状态计算：
   1. 对于不选：\(f(i,j)=f(i-1,j)\) 直接从上个元素对应的状态转移过来即可
   2. 对于加上此元素：\(f(i,0)=\max{(f(i-1,1)+w_i)}\) 加上当前元素必定是从减去上一个元素转移过来
   3. 对于减去此元素：\(f(i,1)=\max{(f(i-1,0)+w_i)}\) 同上

这样我们就可以在\(O(n)\)的复杂度下完成本题的dp计算了

AC代码

#include <bits/stdc++.h>
using namespace std;
constexpr int N = 3e5 + 1000;
typedef long long LL; // 注意爆int
int a[N], n;
LL f[N][2];

int main()
{
    io;
    int _t, q;
    scanf("%d", &_t);
    while (_t --)
    {
        scanf("%d%d", &n, &q);
        for (int i = 1; i <= n; i++) cin >> a[i];
        
        for (int i = 1; i <= n; i++)
        {
            f[i][0] = f[i - 1][0];
            f[i][1] = f[i - 1][1];
            f[i][0] = max(f[i][0], f[i - 1][1] + a[i]);
            f[i][1] = max(f[i][1], f[i - 1][0] - a[i]);           
        }

        cout << max(f[n][0], f[n][1]) << '\n';
    }
    
    return 0;
}

标签：Army,--,元素,不选,int,version,Exit,Pok,dp
来源： https://www.cnblogs.com/FrankOu/p/15436248.html