C-偶遇行军蚁—集训队

代码非本人所写：（仅仅加入个人理解）

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

int main()
{
    string s;
    int n;
    
    // 读入行数和字符串
    cin >> n; // 读入行数
    cin >> s; // 读入字符串
    
    // 定义一个 vector，用于存储按照 Z 字形排列后的字符串
    vector<string> arr(n);
    
    // 如果只有一行，直接输出字符串并结束程序
    if (n == 1)
    {
        cout << s << endl;
        return 0;
    }
    
    int now = 0; // 当前所在行
    int dx = 1; // 行偏移量，用于控制行的上下移动
    
    // 将字符串按照 Z 字形排列存储到 vector 中
    for (int i = 0; i < s.size(); i++)
    {
        arr[now] += s[i]; // 将字符添加到当前行
        now += dx; // 移动到下一行
        // 如果到达了第一行或最后一行，改变行移动方向
        if (now == 0 || now == n - 1)
        {
            dx = -dx; // 行偏移取反，改变移动方向
        }
    }
    
    // 输出 Z 字形排列后的字符串
    for (int i = 0; i < n; i++)
    {
        cout << arr[i];
    }
    
    return 0;
}

样例输入为：

3
PAYPALISHIRING
我们需要将字符串 "PAYPALISHIRING" 按照 Z 字形排列成 3 行。

首先，我们定义了一个大小为 3 的 vector，用于存储按照 Z 字形排列后的字符串：

cpp
vector<string> arr(3);
然后我们进入循环，逐个字符地将字符串中的字符按照 Z 字形排列添加到相应的行中。

循环过程如下：

遍历字符串 "PAYPALISHIRING" 的每个字符。
将字符依次添加到 Z 字形排列的每一行中。
如果到达了 Z 字形排列的边界，则改变行移动方向。
最终得到按照 Z 字形排列后的字符串存储在 arr 中。
接下来我们来逐步看循环过程中的变化：

当 i=0 时，将字符 'P' 添加到第 0 行，现在的 arr 状态为：
P
当 i=1 时，将字符 'A' 添加到第 1 行，现在的 arr 状态为：
P
A
当 i=2 时，将字符 'Y' 添加到第 2 行，现在的 arr 状态为：
P
A
Y
当 i=3 时，将字符 'P' 添加到第 1 行，现在的 arr 状态为：
P
AP
Y
当 i=4 时，将字符 'A' 添加到第 0 行，现在的 arr 状态为：
PA
AP
Y
当 i=5 时，将字符 'L' 添加到第 1 行，现在的 arr 状态为：
PA
APL
Y
以此类推，直到遍历完整个字符串。
最终得到按照 Z 字形排列后的字符串为：

PAHNAPLSIIGYIR

拓展：

#include <iostream>

using namespace std;
const int N=510;
int a[N][N];
    int n;
    int dx[]={0,1,1,-1},dy[]={1,-1,0,1};

int main()
{ 
    cin >> n;
    for(int i=0;i<n;i++)
    for(int j=0;j<n;j++)
    cin >> a[i][j];
    
    int i=0,j=0 ,cnt=0,t=0;
    cout << a[i][j] << ' ';
    cnt++;
    while(cnt<n*n)
    {
        int x=i+dx[t],y=j+dy[t];
        if(a[x][y]&&x>=0&&x<n&&y>=0&&y<n)
        {
            i=x,j=y;
            cout << a[i][j] << ' ';
            a[i][j]=0;
            cnt++;
            if(t==0||t==2)
            t++;
        }
        else
        t=(t+1)%4;
    }
    cout << endl;
    return 0;
    
    
}