原题链接:力扣https://leetcode.cn/problems/beautiful-arrangement-ii/
我的解题思路:
通过推演发现:每当最后一位移动到前方时 整个序列不同的差值绝对值的数量会加2,且起始为1。
例:1 2 3 4 5 这时只有 1 这一种差值 当数列变成 1 5 2 3 4时 此时差值有4 3 1 三种
且当最后两个数字交换时 差值的种类只会增加一种 例 1 5 2 4 3 此时差值有 4 3 2 1四种
(注:我发现在特定情况下,差值的种类数量只会按情况增加,但是差值的种类可能会发生改变,但是我重新找特定情况时 发现找不到了,举例说明,例如原来三种差值为4 3 1,发生一次最后两个数的交换应该是四种,但是这四种可能是5 4 2 1,即原来的3没有了,多了5和2,但是数量只会增加1,所以并不会影响解题)
我的代码:
其中 flag_i 从1开始记录数字 flag_j从最后开始记录数 flag_k记录当前差值种类的数量,
因为题目保证k<n,即题目一定有正解,所以不需要担心这里的while会发生死循环,每往前面移一位差值种类数量+2,当flag_k和k相等或者相差1时,结束循环。相差1 则交换最后两位。相等 则按顺序存储数列。
#include<iostream>
#include<map>
#include<string>
#include<vector>
#include<set>
#include<algorithm>
#include<queue>
#include<unordered_set>
using namespace std;
class Solution {
public:
vector<int> constructArray(int n, int k) {
vector<int>Target;
int flag_i = 1, flag_j = n;
int flag_k = 1;
while (flag_k != k && flag_k != (k - 1))
{
Target.push_back(flag_i);
Target.push_back(flag_j);
flag_i++; flag_j--;
flag_k += 2;
}
if (flag_k == k)
{
for (int i = flag_i; i <= flag_j; i++)
{
Target.push_back(i);
}
}
else
{
for (int i = flag_i; i <= flag_j - 2; i++)
{
Target.push_back(i);
}
Target.push_back(flag_j);
Target.push_back(flag_j - 1);
}
return Target;
}
};
int main()
{
Solution sc;
vector<int>text = sc.constructArray(3, 2);
for (int x : text)
{
cout << x << " ";
}
return 0;
}