【题解】
题意:给定一个n*n大小的网格,上面有两种颜色黑和白,要求使用k*k的橡皮擦擦一次,输出最多的白线(全为白的行和列)数目。
思路:用前缀和维护每个位置作为橡皮左上角的行和列的贡献,最后输出最大贡献即可。
对于每行和每列,我们将执行以下操作(以行为例):
跑出这一行的最左端的黑色方块的位置b和最右端的位置e,如果没有黑色方块,那么对于整个n*n的网格都有贡献+1;如果长度<=k,那么说明可以被橡皮完全覆盖,那么这一行的贡献就要加在子矩阵左上角为(max(0,i-k+1),max(0,e-k+1))到右下角为(i,b)上;否则没有贡献。这个区间如果理解不了,可以想象一下拿一个k*k的正方形在平面上移动,让它始终有覆盖到第i行的b到e的位置,那么这个正方形的左上角的四个方向的边界就很好得到了。
贡献的修改可以直接前缀和,或者通过差分(效率高)实现,如果不会差分的看这里:
最终的p[i][j]保存的是(i,j)这个位置作为橡皮的左上角,对于第i-(i+k-1)行和第j-(j+k-1)列所做出的贡献(如果有)之和。
【直接前缀和(624ms)】
#include<bits/stdc++.h