1. 程式人生 > >leetcode Score After Flipping Matrix

leetcode Score After Flipping Matrix

比較簡單的一題

主要運用貪心的思想

讓權值高的位儘量為1

程式碼如下

class Solution {
public:
    int matrixScore(vector<vector<int>>& A) {
        for(int i=0;i<A.size();i++)
        {
            if(A[i][0]==0)
            {
                for(int q=0;q<A[i].size();q++)
                {
                    if(A[i][q])
                        A[i][q]=0;
                    else
                        A[i][q]=1;
                }
            }
        }
        for(int i=1;i<A[0].size();i++)
        {
            int num=0;
            for(int q=0;q<A.size();q++)
            {
                if(A[q][i])
                    num--;
                else
                    num++;
            }
            if(num>0)
            {
                for(int q=0;q<A.size();q++)
                {
                    if(A[q][i])
                        A[q][i]=0;
                    else
                        A[q][i]=1;
                }
            }
        }
        int result=0;
        for(int i=0;i<A.size();i++)
        {
            int t=0;
            for(int q=0;q<A[i].size();q++)
            {
                t<<=1;
                t+=A[i][q];
            }
            result+=t;
        }
        return result;
    }
};