F
题意:给一个长度为n的由0、1、?构成的串,进行n次游戏。每次游戏需要找出串中有多少组连续的0或者连续的1,其中?既可以看成0,也可以看成1。注意,在每一次游戏时,每个字符只能被统计一次,不能重复出现在不同的组中。
思路:
在每一次游戏中,如果我们能够快速找到下一个含有连续0/1串的位置,就可以在nlogn的时间内统计出答案。
维护
n
x
t
[
0
/
1
]
[
i
]
nxt[0/1][i]
nxt[0/1][i]数组表示下一个0/1到当前位置的距离,维护vector
p
o
s
[
0
/
1
]
[
l
e
n
]
pos[0/1][len]
pos[0/1][len]表示长度至少为len的连续串的。在每一轮游戏中,先判断当前位置向下是否有足够的连续的0/1(即是否在某个0/1串的中间位置),如果有,直接更新到pos+len的位置;如果没有,则需要找下一个新的长度大于len的0/1串,用来更新pos。
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#define LL long long
using namespace std;
const int N = 1e6 + 10;
vector<int>pos[2][N];//用queue炸空间
int nxt[2][N], ptr[2][N];
char s[N];
int n;
void get_pos(int id)
{
for(int l = 1; l <= n;)
{
if(s[l] == '0' + (id ^ 1)) l++;
else
{
int r = l + 1;
while(r < n + 1 && s[r] != '0' + (id ^ 1)) r++;
for(int len = 1; len <= r - l; len ++) pos[id][len].push_back(l);
l = r;
}
}
}
int main()
{
scanf("%d", &n);
scanf("%s", s + 1);
nxt[0][n + 1] = 0;
nxt[1][n + 1] = 0;
for(int i = n; i > 0; i--)
{
if(s[i] != '1') nxt[0][i] = nxt[0][i + 1] + 1;
if(s[i] != '0') nxt[1][i] = nxt[1][i + 1] + 1;
}
get_pos(0);
get_pos(1);
for(int i = 1; i <= n; i++)
{
int cnt = 0;
int now = 1;
while(now <= n)
{
if(nxt[0][now] >= i || nxt[1][now] >= i) now = now + i;
else
{
while(ptr[0][i] < pos[0][i].size() && pos[0][i][ptr[0][i]] < now)
ptr[0][i]++;
while(ptr[1][i] < pos[1][i].size() && pos[1][i][ptr[1][i]] < now)
ptr[1][i]++;
if(ptr[0][i] < pos[0][i].size() && ptr[1][i] < pos[1][i].size())
now = min(pos[0][i][ptr[0][i]], pos[1][i][ptr[1][i]]) + i;
else
{
if(ptr[0][i] >= pos[0][i].size() && ptr[1][i] >= pos[1][i].size())
break;
else
{
if(ptr[0][i] >= pos[0][i].size())
now = pos[1][i][ptr[1][i]] + i;
else
now = pos[0][i][ptr[0][i]] + i;
}
}
}
cnt += 1;
}
cout<<cnt<<" ";
}
cout<<endl;
return 0;
}