CodeForces #642 题解

CodeForces #642

D. Constructing the Array【优先队列】

题目

给出一个长为$n$的全0字符串，要进行$n$次操作，第$i$次操作规则如下：

1. 在当前字符串中找到一个最长且全为0的子串，如果存在多个，选择最左边的一个；
2. 赋值：$s[\frac{l+r}{2}]=i$

求出$n$次操作后得到的字符串

题解

本题中，用结点node表示一段全为0的子串，包括其左边界、有边界和长度

struct node {
	int l, r, len;
	node(int a, int b, int c) { l = a; r = b; len = c; }
	bool operator<(const node&a) const {
		if (len < a.len) return 1;
		if (len == a.len && l > a.l) return 1;
		return 0;
	}
};

每次操作时，需要选取一个最长的结点node1，而操作结束后，这个区间将被一分为二，产生两个新的结点node2,node3，而这两个新的结点的范围可以根据之前的node1来确定，即

int &L = node1.l, &R = node1.r;
int Mid = (L + R) / 2;				// 中间位置
node2(L, Mid - 1, Mid - 1 - L + 1);	// 左半部分
node3(pos + 1, R, R - Mid);			// 右半部分

最后，只需要使用一个优先队列来完成$n$次操作，就能得到最终结果

代码

#include <bits/stdc++.h>
using namespace std;
#define mem(a,b) memset(a,b,sizeof(a))
#define endl '\n'
typedef long long ll;
const int maxn = 2e5 + 100;
int n;
int ans[maxn];
struct node {
	int l, r, len;
	node(int a, int b, int c) { l = a; r = b; len = c; }
	bool operator<(const node&a) const {
		if (len < a.len) return 1;
		if (len == a.len && l > a.l) return 1;
		return 0;
	}
};
void solve()
{
	priority_queue<node>q;
	q.push(node(1, n, n));
	int cnt = 1;
	while (cnt <= n) {
		node now = q.top(); q.pop();
		int mid = (now.l + now.r) / 2;
		ans[mid] = cnt++;
		q.push(node(now.l, mid - 1, mid - 1 - now.l + 1));
		q.push(node(mid + 1, now.r, now.r - mid));
	}
}
int main()
{
	ios::sync_with_stdio(0), cin.tie(0);
	int t; cin >> t;
	while (t--) {
		cin >> n;
		solve();
		for (int i = 1; i <= n; i++) {
			cout << ans[i] << " ";
		}
		cout << endl;
	}
}

E. K-periodic Garland【DP】

题目

给定一个由0,1组成的字符串，要求字符串中任意两个1之间相距为k，求最小操作次数

例如：k=3时，010010,1001001,0010,0都满足条件

• 字符串长度$n\le 10^6$；模$k\le n$

题解

因为最终得到的字符串中，1所在的位置模k都是相同的，因此首先根据模$k$对原字符串进行划分$(i,i+k,i+2k,\dots )$，得到一系列模$k$相等的子序列

对于每一个子序列，将不属于该子序列的1全部变为0，并且将该子序列转换为符合条件的序列（其形式类似于0011100这样，一段连续的0、一段连续的1、再一段连续的0，如此可以保证任意两个1之间的距离一定为$k$），并更新答案。

因此问题就转化为：对于一个字符串，选取一个区间，使得区间里的元素都为1，而区间外的元素都为0
令$dp[i]$表示：假设第$i$位为1的条件下，字符串前$i$位满足条件的最小操作数，其转移方程为

for (int i = 0; i < n; i++) {
    int cur = s[i] - '0';		// 第i位的元素
    pref += cur;				// 记录字符串前i位中有多少个1
    dp[i] = (cur == 0 ? 1 : 0);	// 因为dp[i]是假设第i位是1，所以如果是0，则操作次数需要+1
    // 转移过程：要么根据前一个状态进行转移，要么选择将第i位之前的所有元素全部变为0
    if (i > 0) dp[i] += min(dp[i - 1], pref - cur);
    ans = min(ans, dp[i] + num - pref); //第i位之后的所有1也都要变为0
}

代码

#include <bits/stdc++.h>
using namespace std;
#define mem(a,b) memset(a,b,sizeof(a))
#define endl '\n'
typedef long long ll;
const int maxn = 1e6 + 10;
const int INF = 1 << 30;
int solve(const string &s)
// 选取一个连续的区间，进行操作让区间里的值都成为 1，区间外的值都为 0，并且保证操作次数最小
{
	int n = s.size();
	int num = count(s.begin(), s.end(), '1');
	int ans = num;
	vector<int> dp(n);	// 假设第i位为1的条件下，字符串前i位满足条件的最小操作数
	int pref = 0;
	for (int i = 0; i < n; i++) {
    int cur = s[i] - '0';		// 第i位的元素
    pref += cur;				// 记录字符串前i位中有多少个1
    dp[i] = (cur == 0 ? 1 : 0);	// 因为dp[i]是假设第i位是1，所以如果是0，则操作次数需要+1
    if (i > 0) dp[i] += min(dp[i - 1], pref - cur); // 转移过程
    ans = min(ans, dp[i] + num - pref); //第i位之后的所有1也都要变为0
}
	return ans;
}
int main()
{
	ios::sync_with_stdio(0), cin.tie(0);
	int t;
	cin >> t;
	while (t--) {
		int n, k;
		string s;
		cin >> n >> k >> s;
		vector<string> v(k);	//模k相同的子串
		for (int i = 0; i < n; i++) {
			v[i % k] += s[i];
		}
		int cnt = count(s.begin(), s.end(), '1');// 字符串中1的个数
		int ans = INF;
		for (auto &it : v) {
			// 除本字符串外其他1都要变为0，再加上当前字符串的操作
			int temp = solve(it) + (cnt - count(it.begin(), it.end(), '1'));
			ans = min(ans, temp);
		}
		cout << ans << endl;
	}
	return 0;
}