This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
// #pragma GCC target ("avx,avx2,fma")
// #pragma GCC optimize ("Ofast,inline") // O1 - O2 - O3 - Os - Ofast
// #pragma GCC optimize ("unroll-loops")
#include <bits/stdc++.h>
using namespace std;
#define rep(i, a, b) for (int i = (a); i < (b); ++i)
#define per(i, a, b) for (int i = (b - 1); i >= (a); --i)
#define trav(a, x) for (auto &a : x)
#define all(x) x.begin(), x.end()
#define sz(x) x.size()
#define pb push_back
#define debug(x) cout << #x << " = " << x << endl
#define umap unordered_map
#define uset unordered_set
typedef pair<int, int> ii;
typedef pair<int, ii> iii;
typedef vector<int> vi;
typedef vector<ii> vii;
typedef vector<vi> vvi;
typedef long long ll;
typedef pair<ll, ll> pll;
typedef vector<ll> vll;
typedef vector<pll> vpll;
const int INF = 1'000'000'007;
vi lets[130];
int n, cnt = 0, memo[5001][5001];
string str;
int dp(int pos, int taken) {
if (taken == cnt) return 0;
int &ans = memo[pos][taken];
if (ans != -1) return memo[pos][taken];
ans = INF;
if (lets['e'][taken] <= pos) {
int num_e = upper_bound(all(lets['e']), pos) - lets['e'].begin() - taken;
ans = num_e + (pos - lets['e'][taken]) + dp(lets['e'][taken], num_e + taken);
}
rep(i, 'a', 'z' + 1) {
if (i == 'e') continue;
vi &vec = lets[i];
int idx = upper_bound(all(vec), pos) - vec.begin();
if (idx != sz(vec)) ans = min(ans, 2 + dp(vec[idx], taken));
}
return ans;
}
int main() {
cin.tie(0)->sync_with_stdio(0);
cin.exceptions(cin.failbit);
memset(memo, -1, sizeof(memo));
cin >> n >> str;
rep(i, 0, n) lets[str[i]].pb(i);
cnt = sz(lets['e']);
cout << dp(0, 0) << endl;
return 0;
}
Compilation message (stderr)
vim.cpp: In function 'int dp(int, int)':
vim.cpp:53:17: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
53 | if (idx != sz(vec)) ans = min(ans, 2 + dp(vec[idx], taken));
| ^
vim.cpp: In function 'int main()':
vim.cpp:66:29: warning: array subscript has type 'char' [-Wchar-subscripts]
66 | rep(i, 0, n) lets[str[i]].pb(i);
| ^
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |