题意简化:给定一个只有0和1字符串,可以将该选择一个子串进行循环移动,即最后一位字符移动到第一位,其他字符向后移一位,循环移动的成本为该子串的长度。要让所有的1都排在所有的0后面,求最少的成本。
题解:可以看出,题目的要求就是把在0前的1都往后移直到满足条件,并且选择的串是越小越好的。例如串100,肯定是不会整个选100,应该先选10,变成010,再选10,变成001,遂满足条件。
每个1都要移动到后面,因此我们可以贪心地完成这个过程,从下标1到下标n,只要遇到1在0前面这种情况,就把1移到0后面。同时因为1都被移动到了一起,这会导致1都聚集在一起,出现110这样的情况,此时我们要将两个1都移动到0后,所选择的子串长度就会增加,成本为 c1(1的数量) + 1。
由此可以看出来,我们可以从头到尾遍历字符串,时刻记录1的数量,并且在1的数量不为0时,遇到0字符,就让答案加一次成本即可。
AC代码:
#include <iostream>
#include <map>
#include <set>
#include <string>
#include <vector>
#include <algorithm>
#include <queue>
#include <bitset>
#include <ctime>
using namespace std;
using ll = long long;
ll INF = 4e18;
const int N = 1e6 + 10;
ll a[N], p[N], b[N], c0[N], c1[N], d[N], c[N];
ll dp[N];
ll mod = 1e9 + 7;
bool vis[N];
void solve()
{
string s; cin >> s;
int n = s.length();
s = ' ' + s;
ll ans = 0;
ll c1 = 0;
for (int i = 1; i <= n; i++)
{
if (s[i] == '1')c1++;
else if (s[i] == '0' && c1) ans += c1 + 1;
}
cout << ans << "\n";
}
int main()
{
ios::sync_with_stdio(0), cout.tie(0), cin.tie(0);
int _ = 1; cin >> _;
while (_--)
solve();
return 0;
}