数据结构MOOC

PTA习题

这道题也是相当费事,不过比上一个题好一些,这里我使用了C++的STL库,使得代码量大幅减少。

1、条件准备

这里我准备采用map来存地址和值,因为map的查找效率也是不错的
数组arr是存链表的地址,并且依次相连的。
可能这里没有看懂,等后面看map的实现时应该就明白了。
主函数使cin、cout更快
#include <iostream>
#include <string>
#include <map>
using namespace std;

const int M = 100005;
string arr[M];  //存链表起始地址

int main()
{
    std::ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);   //加速

2、map定义,存储输入数据

这里主要说明一下map:
map<string,pair<int,string>> 可以让我们后续通过输入一个地址就能获取该结点的所有信息。
输入数据每一行为: address  data   next,
address用string这个键来存,data用pair里的int存,next用pair里的string存
 map<string, pair<int, string>> m;
    string qi;//起始地址
    int n;    //总共数据个数
    int k;    //反转数
    cin >> qi >> n >> k;
    for (int i = 1; i <= n; i++)
    {
        string a;
        cin >> a;
        cin >> m[a].first >> m[a].second;
     //Address Data Next为输入格式,Address与Next存为string类型,data用int来存
    }

3、初始化arr数组

这里我们arr数组存的是每一行输入数据的address,循环遍历用过查找本数据的next来使数据连接起来,这样arr数组存的就是首尾相连的数据的address。
string t = qi;  
    int len = 0;   //记录arr数组长度
    while (m.find(t) != m.end())  //直到next为-1停下
    {
        arr[len++] = t;     
        t = m[t].second;   //更新t
    } 
    

4、反转(重点)

此处反转我们考虑恰好能全反转和留有几个小数据不反转的情况。

但无论如何都要先进行反转判断。

 int flag = 0;  //标记,规范输出方式
    for (int i = 1; i * k <= len; i++)
    {
//i代表当前进行第几轮反转,i*k比arr中链表长度短都可以进行
        for (int j = i * k - 1; j >= (i - 1) * k; j--)
        {
       //j表示当前该输出的位置
            if (flag)
                cout << ' ' << arr[j] << endl;    //因为输出的是下一行的地址所以,第一行输出不进行
            cout << arr[j] << ' ' << m[arr[j]].first;  //输出地址和数据
            flag = 1;   //除第一行后面都多输出一个并换行
        }
    }
这里用两重循环,第一重是当前进行第几轮反转,如果轮数与每轮反转数相乘小于arr数组长度即可进入循环体。
第二重是循环该轮反转的位置,从该轮的最后一处倒着遍历,就是i*k-1,直到反转k个结束。
里面flag是不让第一行输出进行的,每一行输出结尾与下一行开头相连,所以j又移动一次才输出该行next。
但是这里有一个问题,如果len==k时恰好输出完,但最后一行next没输出,我们一会再处理这个问题。
下面考虑剩几个数据不够k时不反转输出。

    if (len % k)
    {//取模不为0则证明有数据不反转直接输出
        for (int i = len - len % k; i < len; i++)
        {
         //找到起始位置,一个一个输出
            cout << ' ' << arr[i] << endl;
            cout << arr[i] << ' ' << m[arr[i]].first;
       }
    }
我们通过取模就能判断剩没剩下,用len-len%k找到不反转的起始位置,往后一直输出到结尾。
注意这里循环体内先输出了next和换行,因为刚才说了反转结束后最后一行没输出完,而恰好这组数据有有不反转的,那这里可以处理。
后面输出意思跟前面类似。就是在最后的时候单独处理一下。
最后再把最后一行输出完,是-1直接输出就行
        cout << ' ' << "-1" << endl;

5、总结

这里使用了stl库里map的一些用法,如果不太熟的话可以先去学一下stl

完整代码如下:

#include <iostream>
#include <string>
#include <map>
using namespace std;

const int M = 100005;
string arr[M];

int main()
{
    std::ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);

    map<string, pair<int, string>> m;
    string qi;
    int n;
    int k;
    cin >> qi >> n >> k;
    for (int i = 1; i <= n; i++)
    {
        string a;
        cin >> a;
        cin >> m[a].first >> m[a].second;
    }

    string t = qi;
    int len = 0;
    while (m.find(t) != m.end())
    {
        arr[len++] = t;
        t = m[t].second;
    }

    int flag = 0;
    for (int i = 1; i * k <= len; i++)
    {
        for (int j = i * k - 1; j >= (i - 1) * k; j--)
        {
            if (flag)
                cout << ' ' << arr[j] << endl;
            cout << arr[j] << ' ' << m[arr[j]].first;
            flag = 1;
        }
    }

    if (len % k)
    {
        for (int i = len - len % k; i < len; i++)
        {
            cout << ' ' << arr[i] << endl;
                cout << arr[i] << ' ' << m[arr[i]].first;
        }

    }
        cout << ' ' << "-1" << endl;

    return 0;
}

点赞(0) 打赏

评论列表 共有 0 条评论

暂无评论

微信公众账号

微信扫一扫加关注

发表
评论
返回
顶部