题意:
给你n,k
要求把数字1-n,排序,使得 |p1 - p2|, |p2 - p3|, ..., |pn - 1 - pn| has exactly kdistinct elements.
相邻两个数的差的绝对值 的不同个数 为k
思路:
构造发现
按照 1 n 2 n-1 3 n-2 4 n-3....的顺序构造,会得到恰好n-1个不一样的绝对值
那么如果题目要k个,只需要留下前面k个数(得到k-1个大于1的绝对值),最后把剩下的数 按照升序或者降序排列,就得到一个绝对值1,加起来一共k个
至于是升序还是降序,就看 第k个数 是n/2+1之前的还是之后的。
#include <cstdio>
#include <cmath>
#include <cstring>
#include <string>
#include <algorithm>
#include <iostream>
#include <queue>
#include <map>
#include <set>
#include <vector>
using namespace std;
int tm[100005];
int cmp(int a,int b)
{return a>b;
}
int main()
{
int n,k;
int i ;
scanf("%d%d",&n,&k);
int ok=0;
for (i=1;i<=n/2;i++)
{
tm[++ok]=i;
tm[++ok]=n-i+1;
}
if (n%2)
tm[++ok]=n/2+1;
if (k==n-1)
{
}
else
if (tm[k]<=n/2+1)
sort(tm+k+1,tm+1+n);
else
sort(tm+k+1,tm+1+n,cmp);
for (i=1;i<=n;i++)
{
if (i!=1)
printf(" ");
printf("%d",tm[i]);
}
printf("\n");
return 0;
}