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 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;
}