Bulb Switcher II LeetCode solution in C++

1
class Solution {
2
 public:
3
  int flipLights(int n, int k) {
4
    if (k == 0) return 1;
5
    if (n == 1) return 2;
6
    if (n == 2 && k == 1) return 3;
7
    if (n == 2 || k == 1) return 4;
8
    if (k == 2) return 7;
9
    return 8;
10
  }
11
};