【牛客】[编程题]MP3光标位置C++

1.题目描述

链接https://www.nowcoder.com/questionTerminal/eaf5b886bd6645dd9cfb5406f3753e15

MP3
Player因为屏幕较小，显示歌曲列表的时候每屏只能显示几首歌曲，用户要通过上下键才能浏览所有的歌曲。为了简化处理，假设每屏只能显示4首歌曲，光标初始的位置为第1首歌。

2.思路解析

1. 分情况比较复杂
2. （1）n小于4，那么不用管列表，修改光标即可
3. ①index == 1 && s[i] == ‘U’
4. ②index == n && s[i] == ‘D’
5. （2）n大于4，需要修改列表（用vector实现），修改光标
6. ①index == 1 && s[i] == ‘U’
7. ②index == n && s[i] == ‘D’
8. ③光标在vector的0和3的位置，修改vector
9. 最后打印的时候就打印1到n

3.代码实现

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

int main()
{
    // n歌曲数
    int n = 0;
    // s执行的up或者down操作
    string s;
    while(cin >> n >> s)
    {
        // 光标
        int index = 1;
        // 歌曲数小于4，表示所有的歌曲在同一页
        // 最后打印列表的时候就打印1到n
        if(n <= 4)
        {
            for(int i = 0; i < s.size(); i++)
            {
                // 光标在1，执行up操作
                if(index == 1 && s[i] == 'U')
                    index = n;
                else if(s[i] == 'U')
                    index--;
                // 光标在n，执行Down操作
                else if(index == n && s[i] == 'D')
                    index = 1;
                else if(s[i] == 'D')
                    index++;
             }
            // 打印列表
            for(int i = 0; i < n - 1; i++)
                cout << i + 1 << " "; 
            cout << n << endl;
            // 光标
            cout << index << endl;
        }
        else
        {
            // 表示歌曲列表
            vector<int> table(4);
            // 最开始是1,2,3,4
            for(int i = 0; i < 4; i++)
                table[i] = i + 1;
            for(int i = 0; i < s.size(); i++)
            {
                // 光标在1，执行up操作，先修改光标位置，再修改列表
                if(index == 1 && s[i] == 'U')
                {
                    index = n;
                    for(int i = 0; i < 4; i++)
                        table[3 - i] = n - i;
                }
                // 1.在同一个列表中up
                // 2.光标的位置正好是数组的第一个数字
                else if(s[i] == 'U')
                {
                    if(index == table[0])
                    {
                        // 修改数组的数字
                        for(int i = 0; i < 4; i++)
                        {
                            table[i] -= 1;
                        }
                    }
                    // 修改光标
                    index--;
                }
                else if(index == n && s[i] == 'D')
                {
                    index = 1;
                    for(int i = 0; i < 4; i++)
                        table[i] = i + 1;
                }
                // 1.在同一个列表中up
                // 2.光标的位置正好是数组的第一个数字
                else if(s[i] == 'D')
                {
                    if(index == table[3])
                    {
                        // 修改数组的数字
                        for(int i = 0; i < 4; i++)
                        {
                            table[i] += 1;
                        }
                    }
                    // 修改光标
                    index++;
                }
             }
            // 打印列表
            for(int i = 0; i < 3; i++)
                cout << table[i] << " ";
            cout << table[3] << endl;
            // 打印光标
            cout << index << endl;
        }
    }
    return 0;
}