题解 | 小红的矩阵染色

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;

bool cmp(int a1,int a2)
{
    return a1>a2;
}

int main() {
    //输入
    int n,m,k;cin>>n>>m>>k;
    char c[n][m];
    for(int i = 0;i<n;i++)
     for(int j = 0;j<m;j++)
      cin>>c[i][j];
    //数每段连续的个数，然后排序
    vector<int> v;
    for(int j=0;j<m;j++)
     for(int i=0;i<n;i++)
     {
        int s = 0;
        while(c[i][j]=='o')
        {
            s++;
            i++;
        }
        if(s-1>0)
        v.push_back(s-1);
     }
     sort(v.begin(),v.end(),cmp);
     int ans=0;
     for(auto &i:v)
     {
        if(k>=i+1) ans+=i,k-=i+1;
        if(k<i+1&&k>0) {ans+=k-1;break;}
     }
     cout<<ans;
}