oj.uz

Submission #434964

# Submission time Handle Problem Language Result Execution time Memory
434964 2021-06-22T15:18:59 Z sumit_kk10 Miners (IOI07_miners) C++14
100 / 100
 266 ms 111684 KB 
#include <bits/stdc++.h>
#define fast ios_base::sync_with_stdio(0);cin.tie(NULL);cout.tie(NULL)
#define ll long long int
#define ld long double
using namespace std;
const int N = 1e6 + 5;
const int MOD = 1e9 + 7;
int n, dp[100000][4][4][4][4];
string s;

long long go(int i, int pre1, int pre2, int pree1, int pree2){
    if(i == 0){
        long long ans = 0;
        ans = max(ans, go(i + 1, (s[i] - 'A' + 1), pre1, pree1, pree2) + 1);
        ans = max(ans, go(i + 1, pre1, pre2, (s[i] - 'A' + 1), pree1) + 1);
        return ans;
    }
    if(i == n)
        return 0;
    if(dp[i][pre1][pre2][pree1][pree2] != -1)
        return dp[i][pre1][pre2][pree1][pree2];
    long long ans = 0;
    if(i == 1){
        if((s[i] - 'A' + 1) == pre1)
            ans = max(ans, go(i + 1, (s[i] - 'A' + 1), pre1, pree1, pree2) + 1);
        else if((s[i] - 'A' + 1) != pre1){
            if(pre1 != 0)
                ans = max(ans, go(i + 1, (s[i] - 'A' + 1), pre1, pree1, pree2) + 2);
            else
                ans = max(ans, go(i + 1, (s[i] - 'A' + 1), pre1, pree1, pree2) + 1);
        }
        if((s[i] - 'A' + 1) == pree1)
            ans = max(ans, go(i + 1, pre1, pre2, (s[i] - 'A' + 1), pree1) + 1);
        else if((s[i] - 'A' + 1) != pree1){
            if(pree1 != 0)
                ans = max(ans, go(i + 1, pre1, pre2, (s[i] - 'A' + 1), pree1) + 2);
            else
                ans = max(ans, go(i + 1, pre1, pre2, (s[i] - 'A' + 1), pree1) + 1);
        }
        return dp[i][pre1][pre2][pree1][pree2] = ans;
    }
    if(pre1 == 0 and pre2 == 0)
        ans = max(ans, go(i + 1, s[i] - 'A' + 1, pre1, pree1, pree2) + 1);
    else if(pre2 == 0){
        if(s[i] - 'A' + 1 == pre1)
            ans = max(ans, go(i + 1, s[i] - 'A' + 1, pre1, pree1, pree2) + 1);
        else
            ans = max(ans, go(i + 1, s[i] - 'A' + 1, pre1, pree1, pree2) + 2);
    }
    else{
        if(s[i] - 'A' + 1 == pre1 and s[i] - 'A' + 1 == pre2)
            ans = max(ans, go(i + 1, s[i] - 'A' + 1, pre1, pree1, pree2) + 1);
        else if(s[i] - 'A' + 1 == pre1 or s[i] - 'A' + 1 == pre2)
            ans = max(ans, go(i + 1, s[i] - 'A' + 1, pre1, pree1, pree2) + 2);
        else{
            if(pre1 != pre2)
                ans = max(ans, go(i + 1, s[i] - 'A' + 1, pre1, pree1, pree2) + 3);
            else
                ans = max(ans, go(i + 1, s[i] - 'A' + 1, pre1, pree1, pree2) + 2);
        }
    }
    if(pree1 == 0 and pree2 == 0)
        ans = max(ans, go(i + 1, pre1, pre2, s[i] - 'A' + 1, pree1) + 1);
    else if(pree2 == 0){
        if(s[i] - 'A' + 1 == pree1)
            ans = max(ans, go(i + 1, pre1, pre2, s[i] - 'A' + 1, pree1) + 1);
        else
            ans = max(ans, go(i + 1, pre1, pre2, s[i] - 'A' + 1, pree1) + 2);
    }
    else{
        if(s[i] - 'A' + 1 == pree1 and pree1 == pree2)
            ans = max(ans, go(i + 1, pre1, pre2, s[i] - 'A' + 1, pree1) + 1);
        else if(s[i] - 'A' + 1 == pree1 or s[i] - 'A' + 1 == pree2)
            ans = max(ans, go(i + 1, pre1, pre2, s[i] - 'A' + 1, pree1) + 2);
        else{
            if(pree1 != pree2)
                ans = max(ans, go(i + 1, pre1, pre2, s[i] - 'A' + 1, pree1) + 3);
            else
                ans = max(ans, go(i + 1, pre1, pre2, s[i] - 'A' + 1, pree1) + 2);
        }
    }
    return dp[i][pre1][pre2][pree1][pree2] = ans;
}

void solve(){
    memset(dp, -1, sizeof dp);
    cin >> n >> s;
    for(int i = 0; i < s.size(); ++i){
        if(s[i] == 'M') s[i] = 'A';
        else if(s[i] == 'F') s[i] = 'C';
    }
    cout << go(0, 0, 0, 0, 0) << '\n';
}

int main(){
    fast;
    int t = 1;
    // cin >> t;
    while(t--)
        solve();
    return 0;
}

Compilation message

miners.cpp: In function 'void solve()':
miners.cpp:88:22: warning: comparison of integer expressions of different signedness: 'int' and 'std::__cxx11::basic_string<char>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   88 |     for(int i = 0; i < s.size(); ++i){
      |                    ~~^~~~~~~~~~

Subtask #1
9.0 / 9.0

# Verdict Execution time Memory Grader output
1 Correct 53 ms 100368 KB Output is correct

Subtask #2
9.0 / 9.0

# Verdict Execution time Memory Grader output
1 Correct 50 ms 100472 KB Output is correct

Subtask #3
9.0 / 9.0

# Verdict Execution time Memory Grader output
1 Correct 53 ms 100492 KB Output is correct

Subtask #4
9.0 / 9.0

# Verdict Execution time Memory Grader output
1 Correct 54 ms 100428 KB Output is correct

Subtask #5
9.0 / 9.0

# Verdict Execution time Memory Grader output
1 Correct 51 ms 100424 KB Output is correct

Subtask #6
7.0 / 7.0

# Verdict Execution time Memory Grader output
1 Correct 51 ms 100412 KB Output is correct

Subtask #7
8.0 / 8.0

# Verdict Execution time Memory Grader output
1 Correct 55 ms 100480 KB Output is correct

Subtask #8
8.0 / 8.0

# Verdict Execution time Memory Grader output
1 Correct 58 ms 101012 KB Output is correct

Subtask #9
8.0 / 8.0

# Verdict Execution time Memory Grader output
1 Correct 74 ms 101524 KB Output is correct

Subtask #10
8.0 / 8.0

# Verdict Execution time Memory Grader output
1 Correct 111 ms 103348 KB Output is correct

Subtask #11
8.0 / 8.0

# Verdict Execution time Memory Grader output
1 Correct 205 ms 108968 KB Output is correct

Subtask #12
8.0 / 8.0

# Verdict Execution time Memory Grader output
1 Correct 266 ms 111684 KB Output is correct