P8764 [蓝桥杯 2021 国 BC] 二进制问题

合集 - 蓝桥杯(26)

1.试题 C ：直线2024-02-202.树形动态规划2024-02-253.X 进制减法2024-02-264.后缀表达式2024-02-265.统计子矩阵2024-02-296.包子凑数2024-03-017.整数拼接2024-03-028.地宫取宝2024-03-039.七段码2024-03-0410.砝码称重2024-03-0511.递增三元组2024-03-0712.完全平方数2024-03-0913.[蓝桥杯 2019 省 A] 填空问题 E2024-03-1314.波动数列2024-03-1615.左孩子右兄弟2024-03-1716.接龙数列2024-03-2317.糖果2024-03-2618.蚂蚁感冒2024-03-2819.子串分值2024-03-3020.超级胶水2024-03-3021.P9425 [蓝桥杯 2023 国 B] AB 路线2024-05-1222.P9426 [蓝桥杯 2023 国 B] 抓娃娃2024-05-1223.P9425 [蓝桥杯 2022 国 B] 20222024-05-1324.P8805 [蓝桥杯 2022 国 B] 机房2024-05-1525.P8806 [蓝桥杯 2022 国 B] 搬砖2024-05-16

26.P8764 [蓝桥杯 2021 国 BC] 二进制问题2024-05-19

收起

P8764 [蓝桥杯 2021 国 BC] 二进制问题

一、问题简析

本题采用数位dp求解。

令 \(f[i][j]=\) 在 \(i\) 位二进制中，有 \(j\) 个 \(1\)，共有几个数。（相当于求组合数）
由于数据范围为 \(1\le N\le 10^{18}\)，最大二进制位数设置为 70，防止溢出。

预处理组合数

for (int i = 0; i < MAX; ++i)    f[i][0] = 1;
	for (int i = 1; i < MAX; ++i)
		for (int j = 1; j <= i; ++j)
			f[i][j] = f[i - 1][j - 1] + f[i - 1][j];

分类讨论

从二进制的高位到低位，对于第 \(i\) 位，边界 \(R\) 为 \(a_i\)，记第 \(i\) 位之前有 last 个 \(1\)，分情况：
若 \(a_i==0\)，直接跳过。
若 \(a_i==1\)，

• 第 \(i\) 位放 0，则后 i - 1 位可以放 K - last 个 \(1\)，答案加上 f[i - 1][K - last]。
• 第 \(i\) 位放 1，然后 ++last，接着讨论下一位。

二、Code

#include <bits/stdc++.h>

using namespace std;
typedef long long ll;

ll quickin(void)
{
	ll ret = 0;
	bool flag = false;
	char ch = getchar();
	while (ch < '0' || ch > '9')
	{
		if (ch == '-')    flag = true;
		ch = getchar();
	}
	while (ch >= '0' && ch <= '9' && ch != EOF)
	{
		ret = ret * 10 + ch - '0';
		ch = getchar();
	}
	if (flag)    ret = -ret;
	return ret;
}

const int MAX = 70;
ll N, K, f[MAX][MAX];
int A[MAX]; // 存N的二进制

ll dp(ll x)
{
	if (!x)    return 0; // 0没有1 
	
	// 转换为二进制 
	int cnt = 0; // 二进制位数 
	while (x)
	{
		A[++cnt] = x % 2;
		x /= 2;
	}
	
	ll ans = 0;
	int last = 0; // 第i位之前1的个数 
	for (int i = cnt; i >= 1; --i)
	{
		if (A[i])
		{
			ans += f[i - 1][K - last]; // 第i位放0 
			
			++last; // 第i位放1 
			if (last > K)    break; // 1的个数已大于K，没有接下去讨论的必要 
		}
		
		if (i == 1 && last == K)    ++ans; // 特判，即x本身也满足条件 
	}
	
	return ans;
}

int main()
{
	#ifdef LOCAL
	freopen("test.in", "r", stdin);
	#endif
	
	N = quickin(), K = quickin();
	
	// 组合数初始化
	for (int i = 0; i < MAX; ++i)    f[i][0] = 1;
	for (int i = 1; i < MAX; ++i)
		for (int j = 1; j <= i; ++j)
			f[i][j] = f[i - 1][j - 1] + f[i - 1][j];
	
	cout << dp(N) << endl; // 原式为 dp(N) - dp(0) = dp(N) - 0 
	
	return 0;
}

---

完