[ABC301D] Bitmask

直接贪心

先忽略 $\texttt{?}$，算出一个值 $v$。

如果 $n<v$，那么答案就是 $-1$。

否则 $n \leftarrow n-v$，然后按位贪心，依次选即可。

#include <bits/stdc++.h>
using namespace std;

long long n, val;
string s;

long long ppow(long long a, long long b)
{
	long long res = 1LL;
	for (int i = 1; i <= b; i++) res *= a;
	return res;
}

int main()
{
	cin >> s >> n;
	long long p = s.size() - 1;
	for (int i = 0; i < s.size(); i++)
	{
		if (s[i] == '1')
		{
			val += ppow(2LL, p);
		}
		p--;
	}
	if (n - val < 0)
	{
		printf("-1\n");
	}
	else if (n == val)
	{
		printf("%lld\n", n);
	}
	else
	{
		long long rn = n;
		n -= val;
		p = s.size() - 1;
		for (int i = 0; i < s.size(); i++)
		{
			if (s[i] == '?')
			{
				long long g = ppow(2LL, p);
				if (g <= n)
				{
					n -= g;
				}
			}
			p--;
		}
		printf("%lld\n", rn - n);
	}
	return 0;
}