Codeforces Round #275 (Div. 2) C Diverse Permutation

C Diverse Permutation

题意是，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 100010int 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;}