这题乍一看十分复杂,因为行列都是动态的
但是将行列分开来看,题目顿时就豁然开朗了
对于行
重点在行的计数实际上是一个dp问题,每一行的答案数都可以由上一行(相同列)的答案数转移过来,然而对于每一行,都会由新的一行
eg:
0 | 1 |
1 | * |
* | * |
=》
? | ? |
0 | 1 |
1 |
可以看出,除开新增加的行(最上面一行)行的行答案数实际上是相同的(而上一行有无答案值,则取决于有无新的1在不在第一行),于是我们可以设定一个计数指针idx,遍历input如果出现1,则将其置为m,每次有下一个数进入,都将idx--,如果idx>0,则就将dp[i-1][j]的答案值加一
对于列
每加入一个数,所有列上的数都会向右走,故,每一个不重复的1都会对应使一列合法,因此st记录该列是否有1即可
#include<iostream>
#include<algorithm>
#include<cstring>
#include<string>
#include<vector>
using namespace std;
const int N = 2000006;
bool row[N];
int n, m;
string in;
int main()
{
int _;
cin >> _;
while (_--)
{
cin >> n >> m >> in;
for (int i = 0; i <= m; i++)row[i] = false;
int dp[n + 1][m + 1];
memset(dp, 0, sizeof dp);
int idx = 0, t;
for(register int i = 1;i <= n; i++)
for (register int j = 1; j <= m; j++)
{
t = (i - 1) * m + j - 1;
if (in[t] == '1')idx = m;
dp[i][j] = dp[i - 1][j] + (idx > 0);
idx--;
}
int cnt = 0;
for (register int i = 1; i <= n; i++)
for (register int j = 1; j <= m; j++)
{
t = (i - 1) * m + j - 1;
if (in[t] == '1' && !row[j])
cnt++, row[j] = true;
cout << cnt + dp[i][j] << ' ';
}
puts("");
}
return 0;
}