27
思路:将所有的数二进制展开,构成一个矩阵,通过消元只保留对角线上 1,即求矩阵的秩,最后得到 1 的个数即秩的大小,也就是最少需要的颜色数。
C++实现:
#include <iostream> #include <set> #include <vector> const int N = 8 * sizeof(int);// 整型 int 的二进制位数 using namespace std; void getBinaryMat(const set<int>& s, vector<vector<int> >& bMat) {// 求每个数的二进制,存储在vector里面,最后得到二进制矩阵 if(s.empty()) return; for(set<int>::iterator it=s.begin(); it != s.end(); it++) { vector<int> v(N); int flag = 1; for(int i=0; i < N; i++) { v[i] = (*it) & flag; flag <<= 1; } bMat.push_back(v); } } int getNumOfLeastColors(set<int>& colorSet) {// 求二进制矩阵的秩,即消元,最后看斜对角线上有几个 1 if(colorSet.empty()) return -1; vector<vector<int> > bMat; getBinaryMat(colorSet, bMat); const int Row = bMat.size(); int r = 0, c = 0; for(c=0; c < N && r < Row; c++, r++) {// 从每一列开始,将每一列消到只有 1 个 1 或者全 0 int i = 0; for(i = r; i < Row; i++) {// 寻找这一列第一个非 0 的行 if(bMat[i][c] != 0) break; } if(Row == i) --r; else { swap(bMat[r], bMat[i]); for(int k=i+1; k < Row; k++) { if(0 != bMat[k][c]) { for(int j=c; j < N; j++) { bMat[k][j] ^= bMat[r][j];// 用第 r 行的 1 消除这一列上其他所有行的 1 } } } } } return r; } int main() { int n; while(cin >> n) { set<int> colorSet; int color; for(int i=0; i < n; i++) { cin >> color; colorSet.insert(color); } cout << getNumOfLeastColors(colorSet) << endl; } return 0; }