水题。段尾空字排序,求最k值。最后一行必须算入。
#include <iostream>
#include <algorithm>
using namespace std;
int ma[100007][107];
int n, m, t;
int w[100007];
bool cmp (int a, int b)
{
return a > b;
}
int main()
{
while(~scanf("%d%d%d", &n, &m, &t))
{
for(int i = 0; i < n; i ++)
{
for(int j = 0; j < m; j ++)
{
scanf("%d", &ma[i][j]);
}
}
int sum = n * m;
int tot = 0;
for(int i = 0; i < n; i ++)
{
if(i != n - 1 && (ma[i + 1][0] == 0 && ma[i + 1][1] == 0))
{
int len = 0;
for(int k = m - 1; k >= 0; k --)
{
if(ma[i][k] == 1)
break;
len ++;
}
w[i] = len;
//cout<<len<<endl;
}
for(int j = 0; j < m; j ++)
{
if(ma[i][j] == 1)
{
tot ++;
}
}
}
sort(w, w + n, cmp);
for(int i = 0; i < t - 1; i ++)
{
sum -= w[i];
}
for(int j = m - 1; j >= 0; j --)
{
if(ma[n - 1][j] == 1)
break;
sum --;
}
printf("%d\n", sum - tot- 2 * t);
for(int i = 0; i < n; i ++)
{
w[i] = 0;
for(int j = 0; j < m; j ++)
{
ma[i][j] = 0;
}
}
}
}