OJ Board

Code: Select all

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

typedef vector<string> VS;
typedef map<int, VS> MIS;
typedef MIS::iterator MIt;

int readInt(string s, int &i)
{
    int n = 0;
    while (s[i] != ']')
    {
        n = n*10 + (s[i] - '0');
        i++;
    }
    return n;
}

int reference(vector<int> &v, int ind)
{
    for (int i = 0; i < v.size(); i++)
    {
        if (v[i] == ind) return i + 1;
    }
    v.push_back(ind);
    return v.size();
}

int main()
{
    string line;
    VS text;
    MIS ref;
    vector<int> v;
    while (getline(cin, line))
    {
        if (line.size() > 0 and line[0] == '[')
        {
            int i = 1;
            int ind = readInt(line, i);
            ref[ind].push_back(line.substr(i + 1));
            while (getline(cin, line) and line.size() > 0)
                ref[ind].push_back(line);
        }
        else text.push_back(line);
    }

    int lim = text.size() - 1;
    while (text[lim] == "") lim--;
    int i = 0;
    while (text[i] == "") i++;
    while (i <= lim)
    {
        for (int j = 0; j < text[i].size(); j++)
        {
            if (text[i][j] == '[')
            {
                j++;
                int ind = readInt(text[i], j);
                cout << '[' << reference(v, ind) << ']';
            }
            else cout << text[i][j];
        }
        cout << endl;
        if (text[i] == "")
            while (text[i] == "") i++;
        else i++;
    }

    for (i = 0; i < v.size(); i++)
    {
        cout << endl << '[' << i + 1 << ']';
        for (int j = 0; j < ref[v[i]].size(); j++) cout << ref[v[i]][j] << endl;
    }
}