51nod1790 输出二进制数

题面

题目链接

解题思路

dp1[i][j]表示上次选择到j的区间是[i,j]，则可以向两个方向转移：
1.dp1[i][j+1]表示直接+1到[i,j]这个数字
2.dp1[j+1][j+1+j-i]，表示将[i,j]输出后的结果，需要注意的是如果s[i,j]比s[j+1,j+1+j-i]大，那么应该向dp1[j+1][j+2+j-i]转移。
考虑怎么处理最小操作，我们发现+1操作的次数只会由最终状态决定，因此我们只要记录输出操作的次数即可。
本题卡空间，需要空间复用。
本人的vs编译器递归爆栈，用stack模拟了深度搜索。
复杂度$O(n^2)$。

代码

#include <cstdio>
#include <algorithm>
#include <cstring>
using namespace std;

const int N = 5003;
const int MOD = 1e9 + 7;

int tot = 1, cnt, tp;
int nex[2][N * N / 2], idx[N * (N + 1) / 2], dfn[N * N / 2], sta[N];

void dfs(int u) {
	sta[++tp] = u;
	while (tp) {
		u = sta[tp--];
		dfn[u] = ++cnt;
		if (nex[1][u]) sta[++tp] = nex[1][u];
		if (nex[0][u]) sta[++tp] = nex[0][u];
	}
}

int n;
int *dp1 = nex[0], *dp2 = nex[1];
char str[N];

void add(int &a, int b) {
	a += b;
	if (a >= MOD) a -= MOD;
}

void upd(int &a, int b) {
	if (a > b) a = b;
}

int f(int x, int y) {
	return (1 + y) * y / 2 - y + x;
}

int g(int x, int y) {
	return (n + (n - x + 1)) * x / 2 - (n - y);
}

int main() {
	//freopen("0.txt", "r", stdin);
	scanf("%s", str + 1);
	n = strlen(str + 1);
	for (int i = 1; i <= n; i++) {
		if (str[i] == '0') continue;
		for (int j = i, p = 1; j <= n; j++) {
			int id = str[j] - '0';
			if (nex[id][p] == 0) nex[id][p] = ++tot;
			idx[g(i, j)] = p = nex[id][p];
		}
	}
	dfs(1);
	memset(dp1, 0, sizeof(nex[0]));
	memset(dp2, 0x3f3f3f3f, sizeof(nex[1]));
	dp1[f(1, 1)] = dp2[f(1, 1)] = 1;
	for (int i = 1; i <= n; i++) {
		for (int j = 1; j <= i; j++) {
			int len = i - j;
			if (str[j] == '0' || dp1[f(j, i)] == 0) continue;
			if (i + 1 <= n) {
				add(dp1[f(j, i + 1)], dp1[f(j, i)]);
				upd(dp2[f(j, i + 1)], dp2[f(j, i)]);
			}
			if (str[i + 1] == '0') continue;
			if (i + 1 + len > n) continue;
			if (dfn[idx[g(j, i)]] <= dfn[idx[g(i + 1, i + 1 + len)]]) {
				add(dp1[f(i + 1, i + 1 + len)], dp1[f(j, i)]);
				upd(dp2[f(i + 1, i + 1 + len)], dp2[f(j, i)] + 1);
			}
			else if (i + 2 + len <= n) {
				add(dp1[f(i + 1, i + 2 + len)], dp1[f(j, i)]);
				upd(dp2[f(i + 1, i + 2 + len)], dp2[f(j, i)] + 1);
			}
		}
	}
	int ans1 = 0, ans2 = 0, t1 = 1, s1 = 0; double re = 1, t2 = 1, s2 = 0;
	for (int i = 1; i <= n; i++) re = re * 4;
	for (int i = n; i >= 1; i--)  add(ans1, dp1[f(i, n)]);
	for (int i = n; i >= 1; i--) {
		if (str[i] == '1') s2 += t2, add(s1, t1);
		t1 = t1 * 2 % MOD; t2 = t2 * 2;
		if (str[i] == '0' || dp1[f(i, n)] == 0) continue;
		if (ans2 == 0 || re > s2 + dp2[f(i, n)]) {
			re = s2 + dp2[f(i, n)], ans2 = (s1 + dp2[f(i, n)]) % MOD;
			if (n - i > 1000) break;
		}
	}
	printf("%d\n%d\n", ans1, ans2);
	return 0;
}