Codeforces Round 904 (Div. 2) B. Haunted House
自己写的两个都T了。。没找出问题
思路就是对于i,右面i位中1的pos和为sum1,个数为cnt,那么维护剩余位中最近的cnt个0的pos和为sum0,答案就是sum1-sum0。
jly:被 \(2^i\) 整除即最后 \(i\) 位必须是0。一定是把最右边的1往左移,维护一个区间[l,r)它变成1,后面变成0。l!=r时,r-1那一位一定是1,且是当前最右面的一个1。代码写的好妙。。
#include <cstdio>
#include <vector>
#include <queue>
#include <cstring>
#include <algorithm>
#include <iostream>
#include <stack>
using namespace std;
typedef long long ll;
typedef pair<int, int>P;
const int maxn = 1e5 + 10;
const int INF = 0x3f3f3f3f;
const ll mod = 998244353;
const double eps = 1e-11;
const double pi = 3.141592653;
int main()
{
int t;
cin >> t;
while(t--)
{
int n;
cin >> n;
string s;
cin >> s;
long long ans = 0;
int l = n, r = n;
for(int i = 1; i <= n; ++i)
{
if(ans != -1)
{
if(l > n - i && s[n-i] == '0')
{
l -= 1;
}
else
{
while(l > 0 && s[l-1] == '1') l -= 1;
if(l == 0) ans = -1;
else ans += r - l;
l -= 1;
}
r -= 1;
}
cout << ans << " \n"[i==n];
}
}
return 0;
}