Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- #include<bits/stdc++.h>
- using namespace std;
- int main()
- {
- ios_base::sync_with_stdio(0);
- cin.tie(0);
- string s, tmp = "";
- vector<pair <char, int> > stak;
- vector<string > gg;
- cin >> s;
- s+="<>";
- if(s[0] != '<'){
- s[0] = '<';
- cout << s << endl;
- return 0;
- }
- else if(s[s.size() - 1] != '>'){
- s[s.size() - 1] = '>';
- cout << s << endl;
- return 0;
- }
- stak.push_back(make_pair('<', 0));
- int ind = -1;
- for(int i = 1; i < s.size(); ++i){
- if(s[i] == '<'){
- stak.push_back(make_pair('<',i));
- if(stak.size() > 1){
- ind = stak[1].second;
- break;
- }
- }
- if(s[i] == '>'){
- ind = i;
- if(s[i + 1] != '<'){
- break;
- }
- stak.pop_back();
- }
- }
- bool flag = false, wtf = false;
- for(int i = 1; i < s.size(); ++i){
- if(s[i - 1] == '<'){
- flag = true;
- if(s[i] == '/') wtf = true;
- }
- if(s[i] == '>' && i != ind){
- flag = false;
- if (!wtf){
- gg.push_back(tmp);
- tmp = "";
- }
- else{
- if(gg[gg.size() - 1] == tmp){
- gg.pop_back();
- tmp = "";
- }
- else {
- gg.push_back(tmp);
- break;
- }
- }
- tmp = "";
- wtf = false;
- }
- if((s[i] != '/' || s[i - 1] != '<') && flag){
- tmp += s[i];
- }
- }
- flag = 0;
- for(int i = 1; i < gg[gg.size() - 1].size() - 1; ++i){
- if(gg[gg.size() - 1][i] == '<' || gg[gg.size() - 1][i] == '>'){
- flag = 1;
- }
- }
- tmp = "";
- for(int i = 0; i < s.size(); ++i){
- if(s[i] == '<' || s[i] == '>' || s[i] == '/'){
- if(!flag){
- if(tmp == gg[gg.size() - 2]){
- cout << s.substr(0, i - tmp.size()) + gg[gg.size() - 1] + s.substr(i, s.size() - 2 - i);
- return 0;
- }
- }
- else{
- if(tmp == gg[gg.size() - 1]){
- cout << s.substr(0, i - tmp.size()) + gg[gg.size() - 2] + s.substr(i, s.size() - 2 - i);
- return 0;
- }
- }
- tmp = "";
- }
- else
- tmp += s[i];
- }
- return 0;
- }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement