牛客多校(2020第三场)A Clam and Fish

题目链接：https://ac.nowcoder.com/acm/contest/5668/A

题意：

• 小月有n个单位时间都在钓鱼，每个单位时间有4种状态，有蛤蜊/没蛤蜊，有鱼/没鱼。小月事先知道这n个时间点的状态。每个事件点有四种可能的动作：

1. 若该时间点有鱼，则可以直钓鱼
2. 若该时间点有蛤蜊，则可以利用蛤蜊制造一袋鱼饵。
3. 若该时间点至少有一个鱼饵，则可以利用这个鱼饵钓一条鱼，之后鱼饵数量-1
4. 什么事都不做

• 问小月最多可以获得多少条鱼

题解：

• 平台有鱼就钓鱼
• 如果有蛤蜊就拿来做鱼饵，没蛤蜊就用鱼饵钓鱼，如果最后还有x包鱼饵，那么就把制作比较晚的x/2包鱼饵的时间拿来钓鱼

#include<iostream>
#include<cstring>
#include<algorithm>
#include<string>
using namespace std;

const int MAX_N = 2 * 10E6 + 1;
int n;
int num_01[MAX_N];
string s;

void solve() {
    int sum_fish = 0, sum_food = 0;
    
    for (int i = 0; i < n; i++) {
        switch(s[i]) {
            case '0': {
                if (sum_food != 0) {
                    sum_food--;
                    sum_fish++;
                }
                break;
            }
            case '1': {
                sum_food++;
                break;
            }
            case '2':
            case '3': {
                sum_fish++;
                break;
            }
        }
    }
    if (sum_food) {
        sum_fish += sum_food/2;
    }
    cout << sum_fish << "
";
    return;
}

int main() {
    int t;
    cin >> t;
    while (t--) {
        cin >> n;
        cin >> s;
        solve();
    }
    return 0;
}

附上之前傻傻ac做法

#include<iostream>
#include<cstring>
#include<algorithm>
#include<string>
using namespace std;

const int MAX_N = 2 * 10E6 + 1;
int n;
int num_01[MAX_N];
string s;

void solve() {
    int sum_fish = 0, sum_food = 0;
    int num_3 = count(s.begin(), s.end(),'3');
    int num_2 = count(s.begin(), s.end(),'2');
    sum_fish = sum_fish + num_2 + num_3;
    
    for (int i = 0; i < n; i++) {
        switch(s[i]) {
            case '0': {
                if (sum_food != 0) {
                    sum_food--;
                    sum_fish++;
                }
                break;
            }
            case '1': {
                if (sum_food >= num_01[i] - 1 && sum_food != 0) {
                    sum_fish++;
                    sum_food--;
                }
                else {
                    sum_food++;
                }
                break;
            }
            default : break;
        }
    }

    cout << sum_fish << "
";
    return;
}

int main() {
    int t;
    cin >> t;
    while (t--) {
        memset(num_01, 0, sizeof(num_01));
        cin >> n;
        s = "";
        cin >> s;
        num_01[0] = count(s.begin(), s.end(),'0');
        num_01[0] += count(s.begin(), s.end(), '1');
        int temp = 1;
        for (int i = 1; i < n; i++) {
            if (s[i] == '0' || s[i] == '1') {
                num_01[i] = num_01[0] - temp;
                temp++;
            }
        }
        solve();
    }
    return 0;
}