LEETCODE 48 672. Bulb Switcher II

来源:互联网 发布:芜湖一中王海飞 知乎 编辑:程序博客网 时间:2024/06/02 02:46
/*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();    }};