Codeforces Round 904 (Div. 2)

发布时间 2023-12-03 02:12:00作者: .Ivorelectra

B. Haunted House

自己写的两个都T了。。没找出问题
思路就是对于i,右面i位中1的pos和为sum1,个数为cnt,那么维护剩余位中最近的cnt个0的pos和为sum0,答案就是sum1-sum0。
jiangly:被 \(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;
}