二分/貪心(ICPC小米預賽第一場 A 2020)

KalznAsawind發表於2020-10-31

題目連結
題意:給你一個只包含0,1,2的序列,問你最多可以同時取出多少個“2020”子序列。
題解:比賽時是隊友寫出的,讚歎隊友的思維能力,真的強。二分答案,然後貪心取即可。檢查時傳入需要組成x個,遍歷陣列,每次遇到2,即優先讓沒有開始組成的序列獲取,然後由已經組成到“20”的序列獲取。每次遇到0,每次優先由已經組成“2”的序列獲取,然後由已經組成“202”的序列獲取。最後檢視是否組成了x個“2020”
下面是ac程式碼:

// % everyone
#include <cstdio>
#include<iostream>
#include<cstring>
#include <map>
#include <queue>
#include <set>
#include <cstdlib>
#include <cmath>
#include <algorithm>
#include <vector>
#include <string>
#include <list>
#include <cctype>
#include <time.h>

namespace IO {
    double start_time = 0.0;
    void ct() { start_time = clock(); return; }
    void fast_cin() { std::ios::sync_with_stdio(false); std::cin.tie(); }
    void read_f(int flag = 0) { freopen("0.in", "r", stdin); if(!flag) freopen("0.out", "w", stdout); }
    void run_time() { std::cout << "\nESC in : " << ( clock() - start_time ) * 1000.0 / CLOCKS_PER_SEC << "ms" << std::endl; }
}
using namespace IO;
template <typename T>
bool bacmp(const T & a, const T & b) { return a > b; }
template <typename T>
bool pecmp(const T & a, const T & b) { return a < b; }

#define ll long long
#define ull unsigned ll
#define _min(x, y) ((x)>(y)?(y):(x))
#define _max(x, y) ((x)>(y)?(x):(y))
#define max3(x, y, z) ( max( (x), max( (y), (z) ) ) )
#define min3(x, y, z) ( min( (x), min( (y), (z) ) ) )
#define pr make_pair
#define pb push_back
using namespace std;

const int N = 1e6+5;
const int inf = 0x3f3f3f3f;
const int mod = 1e9+7;

char su[N];
int dp[6];
int n;
bool che(int x)
{
    memset(dp, 0, sizeof(dp)); dp[0] = x;
    for (int i = 1; i <= n; i++)
    {
        if (su[i] == '2')
        {
            if (dp[0]) dp[0]--,dp[1]++;
            else if (dp[2]) dp[2]--,dp[3]++;
        }
        else if (su[i] == '0')
        {
            if (dp[1]) dp[1]--,dp[2]++;
            else if (dp[3]) dp[3]--,dp[4]++;
        }
    }
    return dp[4]==x;
}
int main()
{

    while(scanf("%d", &n) != EOF)
    {
        scanf("%s", su+1);
        int n = strlen(su+1);
        int l = 0, r = n;
        while(l <= r)
        {
            int mid = (l + r) >> 1;
            if (che(mid)) l = mid+1;
            else r = mid-1;
        }
        printf("%d\n", l-1);
    }
}

相關文章