这个有意思,收录一下
D - Gomamayo Sequence (atcoder.jp)
#include <iostream>
#include <string>
#include <stack>
#include <vector>
#include <queue>
#include <deque>
#include <set>
#include <map>
#include <unordered_map>
#include <unordered_set>
#include <algorithm>
#include <cstdlib>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <numeric>
#include <functional>
#include <climits>
//#include <windows.h>
using namespace std;
#define ll long long
#define db double
#define PII pair<int, int>
#define TUP tuple<int, int, int>
const int inf = 0x3f3f3f3f;
ll mod = 1e15 + 17;
ll q_pow(ll a, ll b) { ll ret = 1; while (b) { if (b & 1)ret = ret * a % mod; b >>= 1; a = a * a % mod; }return ret; }
ll gcd(ll a, ll b) { if (!b)return a; return gcd(b, a % b); }const int N = 1e6;int n;
string s;
ll w[N];
ll sum01[N], sum10[N];
void yelp_solve() {cin >> n >> s;for (int i = 0; i < n; i++) cin >> w[i];sum01[0] = s[0] == '0' ? 0 : w[0];for (int i = 1; i < n; i++) {if (i % 2 == 0) {sum01[i] = sum01[i - 1] + (s[i] == '1' ? w[i] : 0);}else {sum01[i] = sum01[i - 1] + (s[i] == '0' ? w[i] : 0);}}sum10[0] = s[0] == '1' ? 0 : w[0];for (int i = 1; i < n; i++) {if (i % 2 == 0) {sum10[i] = sum10[i - 1] + (s[i] == '0' ? w[i] : 0);}else {sum10[i] = sum10[i - 1] + (s[i] == '1' ? w[i] : 0);}}ll ans = LLONG_MAX;for (int i = 0; i + 1 < n; i++) {if (i % 2 == 0) {//00ans = min(ans, (i - 1 >= 0 ? sum01[i - 1] : 0) + sum10[n - 1] - sum10[i + 1] + (s[i] == '0' ? 0 : w[i]) + (s[i + 1] == '0' ? 0 : w[i + 1]));//11ans = min(ans, (i - 1 >= 0 ? sum10[i - 1] : 0) + sum01[n - 1] - sum01[i + 1] + (s[i] == '1' ? 0 : w[i]) + (s[i + 1] == '1' ? 0 : w[i + 1]));}else {//00ans = min(ans, (i - 1 >= 0 ? sum10[i - 1] : 0) + sum01[n - 1] - sum01[i + 1] + (s[i] == '0' ? 0 : w[i]) + (s[i + 1] == '0' ? 0 : w[i + 1]));//11ans = min(ans, (i - 1 >= 0 ? sum01[i - 1] : 0) + sum10[n - 1] - sum10[i + 1] + (s[i] == '1' ? 0 : w[i]) + (s[i + 1] == '1' ? 0 : w[i + 1]));}}cout << ans;return;
}
int main() {int r = 1;//cin >> r;while (r--) yelp_solve();return 0;
}