SCPC ：RGB (思维 / 线段树/求最大区间和)

传送门

题意：
给你一个字符串。
字符串由：‘R’ , ‘G’ , ‘B’ 组成。
找一个区间，使得能获得的价值最大。
价值计算公式：5 * r - 3 * g + b
r , g , b 分别为字符 R , G , B 的个数。

输入：
第一行输入一个 N 为字符串长度。
第二行输入一个长度为 N 的字符串。
0 <= N <= 1000000

输出：
输出能获得的最大价值。

输入样例 1
3
RBG
输出样例 1
6
提示
选择区间[ 1 , 2 ] 价值最大为 5 + 1 = 6.

思路1： 简单思维

• 一个R等价于5，那么一个B和一个G就分别等价于1和-3
• 直接用一个sum统计区间和，一遇到sum < 0就将sum重置为0，期间ans不点和sum取max就行啦。

其实我以前写过这个题，但是比赛的时候竟然没想到这种方法反而选择了线段树，又被各种小错误逼疯，嗐~

代码实现：

#include<bits/stdc++.h>
using namespace std;
const int N = 1e6 + 5;
int n, x, sum, ans;
char a[N];

int main()
{
  cin >> n >> a;
  int len = strlen(a);
  for(int i = 0; i < len; i ++){
    if(a[i] == 'R') x = 5;
    else if(a[i] == 'B') x = 1;
    else x= -3;
    if(sum < 0) sum = 0;
    sum += x;
    ans = max(ans, sum);
  }
  cout << ans << endl;
  return 0;
}

思路2: 线段树

• 既然是求最大区间和，自然可以用线段树来做，这里就可以直接参考这个题啦。
• 注意：线段树的时候需要特判一下N为0的时候输出为0，不然就像我一样，一直运行错误。

比赛的时候用的char数组，然后好几次都把strlen()函数放在了for循环里面，导致一直超时，嗐脑抽一直找不到错(此次深刻反思)。

代码实现：

#include<cstdio>
#include<cstring>
#include<algorithm>

using namespace std;
const int N = 1e6 + 5;

int n, m;
char s[N];
int w[N];

struct node
{
    int l, r;

    int sum, lmax, rmax, tmax;
}tr[N << 2];

void pushup(node &u, node &l, node &r)
{
    u.sum = l.sum + r.sum;

    u.lmax = max(l.lmax, l.sum + r.lmax);
    u.rmax = max(r.rmax, r.sum + l.rmax);
    u.tmax = max(max(l.tmax, r.tmax), l.rmax + r.lmax);
}

void pushup(int u)
{
    pushup(tr[u], tr[u << 1], tr[u << 1 | 1]);
}

void build(int u, int l, int r)
{

    if(l == r){
        tr[u] = {l, r, w[r], w[r], w[r], w[r]};
        return;
    }
    else{
        tr[u] = {l, r};
        int mid = l + r >> 1;
        build(u << 1, l, mid);
        build(u << 1 | 1, mid + 1, r);
        pushup(u);
    }
}

int main()
{
    scanf("%d", &n);
    getchar();
  
    scanf("%s", s);
    int len = strlen(s);
  
    for(int i = 0; i < len; i ++){
        if(s[i] == 'R') w[i] = 5;
        else if(s[i] == 'B') w[i] = 1;
        else w[i] = -3;
    }

    build(1, 0, n - 1);
    printf("%d\n", tr[1].tmax);
    return 0;
}