题意是,1到n的全排列,记为
p1, p2, ..., pn
.,让你找出一种排列,这种排列满足
|p1 - p2|, |p2 - p3|, ..., |pn - 1 - pn|
,这n-1个数有k个不同的数。其中1 ≤ k < n ≤ 105
我是用构造的方法做的,1到k+1的全排列一定有一个可以使得上式有k个不同的数,为1到k,我们使第一个数为1,|pi - pi+1|等于k-i+1,就可以构造出来,然后k+2到n依次排列就可以了,只有|pk+1 - pk+2|这个地方才可能超过k,其他地方都是1,因为第k+2个数是k+2,所以第k+1个数为1时才能超过k,但是1已经放在了第一个位置,所以这个数一定小于等于k。这就构造出来了答案。
但我在写的时候没有考虑前k+1个数必须是1到k+1的排列,我忽略了这个条件,错了好多次,最后才发现了。
代码如下:
#include <cstdio>
#include <cstring>
#define N 100010
int ans[N], n, k, vis[N];
int main(void)
{
while (scanf("%d%d", &n, &k) != EOF) {
memset(ans, 0, sizeof(ans));
memset(vis, 0, sizeof(ans));
ans[1] = 1;
vis[1] = 1;
for (int i = 2; i <= k + 1; ++i) {
int m = k - i + 2;
if (ans[i - 1] + m > k + 1) {
ans[i] = ans[i - 1] - m;
} else {
int tmp = ans[i - 1] + m;
if (vis[tmp] == 0) {
ans[i] = ans[i - 1] + m;
} else {
ans[i] = ans[i - 1] - m;
}
}
vis[ans[i]] = 1;
}
int cnt = k + 2;
for (int i = cnt; i <= n; ++i) {
ans[i] = i;
}
for (int i = 1; i <= n; ++i) {
printf("%d%c", ans[i], (i == n) ? '\n' : ' ');
}
}
return 0;
}