Q5.1 Set Bits

最新推荐文章于 2024-07-19 14:21:55 发布

「已注销」

最新推荐文章于 2024-07-19 14:21:55 发布

阅读量469

点赞数

CC 4.0 BY-SA版权

分类专栏： Cracking the coding interview 文章标签： Bit Manipulatin

本文链接：https://blog.youkuaiyun.com/xinxinxv/article/details/42977197

Cracking the coding interview 专栏收录该内容

50 篇文章

订阅专栏

给定32位数N和M，以及位位置i和j，设计方法将N中从i到j的所有位设置为M的值。例如，M成为N位于i处并从j开始的子串。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

You are given two 32-bit numbers, N and M, and two bit positions, i and j. Write a method to set all bits between i and j in N equal to M (e.g., M becomes a substring of N located at i and starting at j).

EXAMPLE:

Input: N = 10000000000, M = 10101, i = 2, j = 6

Output: N = 10001010100

先将N的第0为到i位保存到ret，不包括第i位，接下来将N的第0位到第j位置0，包括第j位。最后或上(m<<i) | ret。

#include <iostream>
#include <vector>
using namespace std;

void print_binary(int n) {
	vector<int> bit;
	int m = 1;
	int len = 8*sizeof(int);
	while(len--) {
		if (n&m) bit.push_back(1);
		else bit.push_back(0);
		m <<= 1;
	}
	while (!bit.empty()) {
		cout<<bit.back();
		bit.pop_back();
	}
	cout<<endl;
}

int update_bits(int n, int m, int i, int j) {
	int ret = (1<<i)-1;
	ret &= n;
	return (n>>(j+1))<<(j+1) | (m<<i | ret);
}

int main(){
    int n = 1<<10, m = 21;
    int ans = update_bits(n, m, 2, 6);
    print_binary(n);
    print_binary(m);
    print_binary(ans);
    return 0;
}