write function to set ith bit in cpp

justanindieguy · Jan 24, 2023 · ea31fa1 · ea31fa1
1 parent 8ae04d1
commit ea31fa1
Showing 1 changed file with 24 additions and 0 deletions.
diff --git a/Bit-manipulationTechniques/SetIthBit/main.cpp b/Bit-manipulationTechniques/SetIthBit/main.cpp
@@ -0,0 +1,24 @@
+/*
+ * Write a function to:
+ *   - set ith bit
+ */
+#include <iostream>
+
+using namespace std;
+
+void setIthBit(int &n, int i)
+{
+    int mask = (1 << i);
+    n = (n | mask);
+}
+
+int main()
+{
+    int i, n = 5;
+    cin >> i;
+
+    setIthBit(n, i);
+    cout << "New value of n: " << n << endl;
+
+    return 0;
+}