【每日一题】ABC301D - Bitmask | 贪心 | 简单

题目内容

原题链接

给定一个长度在 $1$ 到 $60$ 之间字符串，字符串取值为 '1', '0', '?' ，一个正整数 $n$ 。

现在将 '?' 修改为 '1' 或者 '0' ，使得 $01$ 串对应的十进制数 $x$ 尽可能大，但是 $x\le n$ ，如果不能做到，输出 $-1$，否则输出 $x$ 。

数据范围

• $s_i\in {[}^{\prime }{0}^{\prime }{,}^{\prime }{1}^{\prime }{,}^{\prime }{?}^{\prime }]$
• $1\le len(s)\le 60$
• $1\le n\le 10^{18}$

题解

从高位到低位贪心修改即可，如果高位能修改为 '1' ，则修改为 ’1‘ ，否则修改为 '0' ，如果都为 '0' 的情况下仍然不能使得结果小于等于 n ，则输出 $-1$ 。

代码

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

int main()
{
    ios::sync_with_stdio(false);
    cin.tie(nullptr);

    string s;
    cin >> s;

    long long x;
    cin >> x;

    int n = int(s.size());
    long long val = 0;
    for (int i = 0; i < n; ++i) {
        if (s[i] != '?') {
            int b = s[i] - '0';
            if (b) val |= 1ll << (n - 1 - i);
        }
    }

    if (val > x) {
        cout << "-1\n";
        return 0;
    }

    for (int i = 0; i < n; ++i) {
        if (s[i] == '?') {
            if (val + (1ll << (n - 1 - i)) <= x) {
                val += 1ll << (n - 1 - i);
                s[i] = '1';
            } else {
                s[i] = '0';
            }
        }
    }

    cout << val << "\n";

    return 0;
}