/*
2017-09-03
看完大佬的代码发现这题其实非常简单粗暴的思路
实际上每一种方案都是做或者不做,总共16种方案
于是我们枚举这16种方案,先验证是否可行,然后由于n长度的问题,不一定每一种方案都是不同的结果,所以再把所有方案实施一遍放到set里。
*/
#define change s[i]=='0'?'1':'0'
class Solution {
public:
int flipLights(int n, int m) {
set<string> st;
for(int i=0;i<16;i++) //用二进制来枚举16种方案
{
int sum_of_one=0; //统计二进制中1的数量
int ii=i;
while(ii) {
if(ii&1) sum_of_one++;
ii/=2;
}
if(sum_of_one%2!=m%2||sum_of_one>m) continue;
string s(n,(char)'0');
// cout<<i<<endl;
if(i&1) for(int i=0;i<n;i++) s[i]=change;
if(i&2) for(int i=0;i<n;i+=2) s[i]=change;
if(i&4) for(int i=1;i<n;i+=2) s[i]=change;
if(i&8) for(int i=0;i<n;i+=3) s[i]=change;
st.insert(s);
}
return st.size();
}
};
LEETCODE 48 672. Bulb Switcher II
最新推荐文章于 2019-05-18 19:59:50 发布