编写函数，将一个数的指定位置置0或置1

//编写函数，将一个数的指定位置置0或置1
#include <stdio.h>
unsigned int set_bit(unsigned int num, int pos, int flag)
{
	int n = 1;
	n = n << (pos - 1);    //将n的第pos位置1，其它全为0
	if (flag == 0)
	{
		num = num&(~n);
	}
	else if (flag == 1)
	{
		num = num | n;
	}
	else
		printf("flag error\n");
	return num;
}
int main()
{
	printf("%d\n", set_bit(3,1,0));     //2
	printf("%d\n", set_bit(4, 2, 1));   //6
	printf("%d\n", set_bit(15, 5, 1));  //31
	return 0;
}