SRM457 Div2 Easy(250) TheSquareDivTwo

TheSquareDivTwo

#include <string>
#include <vector>
#include <algorithm>

using namespace std;

class TheSquareDivTwo
{
public:
    vector <string> solve( vector <string> board );
};

vector <string> TheSquareDivTwo::solve( vector <string> board )
{
    int n = (int)board.size();

    vector<int> R;
    for ( int i=0; i<n; i++ )
        R.push_back( count(board[i].begin(),board[i].end(),'C') );

    for ( int i=0; i<n; i++ )
    for ( int j=0; j<n; j++ )
        board[i][j] = i<n-R[j] ? '.' : 'C';

    return board;
}