链接:https://ac.nowcoder.com/acm/contest/63869/C
来源:牛客网
题目描述
小美定义一个 01 串的权值为:每次操作选择一位取反,使得相邻字符都不相等的最小操作次数。例如,"10001"的权值是 1,因为只需要修改一次:对第三个字符取反即可。
现在小美拿到了一个 01 串,她希望你求出所有非空连续子串的权值之和,你能帮帮她吗?
输入描述:
一个仅包含'0'和'1'的字符串,长度不超过 2000。输出描述:
所有非空子串的权值和。 示例1输入
10001
输出
8
说明
长度为 2 的子串中,有 2 个"00"的权值是 1。 长度为 3 的 3 个子串权值都是 1。 长度为 4 的 2 个子串权值都是 1。 长度为 5 的 1 个子串权值是 1。 总权值之和为 2+3+2+1=8思路:
01字符串s .....,s[i-1],s[i],.....
首先我们定义f[n][2],其中[2]表示状态位,也就是当前第i位变与不变(0表示第i位不变,1表示第i位变)
f[i][0]、f[i][1]其含义分别为以第i个字符结尾的字符串,第i位变的权值和第i位不变的权值
为了缩小问题规模,考虑第i位和第i-1位的情况:
-
若s[i]==s[i-1],此时i和i-1这两个之间必须要变一个,可以分为以下两种情况:i不变i-1变;i变i-1不变,状态转移方程如下
-
f[i][0] = f[i-1][1];
-
f[i][1] = f[i-1][0] + 1;
-
-
若s[i]!=s[i-1],此时也可以分为两种情况:i和i-1都不变;i和i-1都变,状态转移方程如下
-
f[i][0] = f[i-1][0];
-
f[i][1] = f[i-1][1] + 1;
-
初始状态定义为f[i-1][0] = 0;f[i-1][1] = 1; // 1表示当前为修改所以初始值为1
第i位变与不变的操作次数都能满足要求,由于题目要求选择操作次数最小,所以记录答案时取min(f[i][0],f[i][1]);
然后我们只需要枚举起始下标,记录每次不同起始下标所能求得的权值之和,最后相加即可
我们每次操作都只用到了当前位i和前一位i-1的值,所以对于f[n][2]数组来说我们只要保证当前位的前一位i-1的值是正常的即可,后续操作数据可直接覆盖也无需初始化Code
#include <bits/stdc++.h> using namespace std; int main() { string s; cin >> s; int n = s.size(); int f[2001][2]; long long ans = 0; for (int j = 0; j < n; j++) { long long sum = 0; f[j][0] = 0, f[j][1] = 1; for (int i = j + 1; i < n; i++) { if (s[i] == s[i - 1]) { f[i][0] = f[i - 1][1]; f[i][1] = f[i - 1][0] + 1; } else { f[i][0] = f[i - 1][0]; f[i][1] = f[i - 1][1] + 1; } sum += min(f[i][0], f[i][1]); } ans += sum; } cout << ans << endl; return 0; }
标签:子串,周赛,01,int,美的,long,权值,位变 From: https://www.cnblogs.com/odd-ryj/p/17665212.html