柠檬微趣北交通9月5日线下笔试真题

柠檬微趣北交通9月5日线下笔试真题

1、问题描述

​ 给定一个一维数组a,长度为n*(n+1)/2,能够正好表现为一个正三角形，如（1），现在，将这个正三角形进行旋转，得到（2），要求原地修改数组a。

  1              4
 2 3            5 2
4 5 6          6 3 1
 (1)            (2)
 原数组a=[1,2,3,4,5,6]
 修改后a=[4,5,2,6,3,1]

2、解题思路

​ 将正三角形进行旋转的过程中，不仅需要三角形边上的各个元素进行旋转，当n过大时，三角形内部还会嵌套一些小三角形，想要获取这些小三角形的位置较为困难。而且原题将一个一维数组抽象为二维的正三角形，由于正三角形的形状特殊，想要直接找到旋转规律也比较困难，因此若是将这个正三角形看做是一个直角三角形，就能够用二维坐标快速寻找到各个元素在一维坐标中的位置。例如，我们将水平向右的方向为y轴正方向，以竖直向下方向为x轴正方向，则a[0]的坐标就是（0,0），a[3]的坐标就是(2,0)，a[5]的坐标就是(2,2)。

​ 然后调换这三个点的坐标，以此形式将三角形边上的所有点位进行调换。考虑到可能有小三角形的存在，需要对内部的小三角形也以此形式进行排序，因此需要得到小三角形的坐标，如（2），5相对于1，x坐标加2，y坐标+1；8相对于11,x坐标减1，y坐标加1；9相对于15，x坐标减1，y坐标减2；然后得到小三角形的位置，然后交换。当小三角形的边长为1时，所有元素都已被交换。

​

1
2 3
4 5 6
（1）

1
2  3
4  5  6               5
7  8  9  10           8 9
11 12 13 14 15   
（2）

3、代码实现

现在进行代码实现

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;

int N;
//打印三角形的信息
void print(int *a, int n)
{
    int count = 0;
    for (int i = 0; i < n; ++i)
    {
        for (int j = 0; j <= i; ++j)
        {
            cout << a[count++] << ' ';
        }
        cout << endl;
    }
    cout << endl;
}

//用于交换的函数
void rotate_help(int *a, int x1, int x2, int y1, int y2, int z1, int z2)
{
    a[x1 * (x1 + 1) / 2 + x2] = a[x1 * (x1 + 1) / 2 + x2] ^ a[y1 * (y1 + 1) / 2 + y2];
    a[y1 * (y1 + 1) / 2 + y2] = a[x1 * (x1 + 1) / 2 + x2] ^ a[y1 * (y1 + 1) / 2 + y2];
    a[x1 * (x1 + 1) / 2 + x2] = a[x1 * (x1 + 1) / 2 + x2] ^ a[y1 * (y1 + 1) / 2 + y2];

    a[y1 * (y1 + 1) / 2 + y2] = a[y1 * (y1 + 1) / 2 + y2] ^ a[z1 * (z1 + 1) / 2 + z2];
    a[z1 * (z1 + 1) / 2 + z2] = a[y1 * (y1 + 1) / 2 + y2] ^ a[z1 * (z1 + 1) / 2 + z2];
    a[y1 * (y1 + 1) / 2 + y2] = a[y1 * (y1 + 1) / 2 + y2] ^ a[z1 * (z1 + 1) / 2 + z2];
}

//通过递归实现各个三角形中各个点位的交换
void rotate(int *a, int x1, int x2, int y1, int y2, int z1, int z2)
{
    //当边长为1是，n=0
    int n = z2 - y2;
    //cout << "n==" << n << endl;
    if (n <= 0)
        return;

    //每次交换的坐标点位
    int new_x1 = x1;
    int new_x2 = x2;
    int new_y1 = y1;
    int new_y2 = y2;
    int new_z1 = z1;
    int new_z2 = z2;

    //每一个三角形要交换几次
    for (int i = 0; i < n; ++i)
    {
        rotate_help(a, new_x1, new_x2, new_y1, new_y2, new_z1, new_z2);
        //print(a, N);
        
        //更新要交换的点位
        new_x1++;
        new_x2++;
        new_y1--;
        new_z2--;
    }
    //外围的三角形交换完后，交换小三角形
    rotate(a, x1 + 2, x2 + 1, y1 - 1, y2 + 1, z1 - 1, z2 - 2);
}

int main()
{

    cin >> N;
    int size = N * (N + 1) / 2;
    int *a = new int[size];
    for (int i = 0; i < size; ++i)
    {
        a[i] = i;
    }
    print(a, N);
    // rotate_help(a, 0, 0, N - 1, 0, N - 1, N - 1);
    rotate(a, 0, 0, N - 1, 0, N - 1, N - 1);
    print(a, N);
}

4、总结思考

代码的实现其实并不算复杂，难点在于能否想到以直角三角形的方式来获取正三角形的点位信息，建立一维与二维的联系，从而在原数组中进行点位的交换。