Нужно оптимизировать код

#include<iostream>
#include<set>
#include<string>
#include <sstream>     
#include <algorithm>   
#include<vector>
using namespace std;
int main(int argc, char* argv[])
{
    set<int> main,tmp,yes,no;
    
    int max;
 
    string str;
    cin>>max;
 
    for(int i = 1; i<=max;i++)
        yes.insert(i);
 
    while(str!="HELP"){
        cin>>str;
        istringstream iss (str);
        for(int i = 0;i<str.size();i++)
        {
            int x;
            iss>>x;
            if (x <= max)
                tmp.insert(x);
        }
        if (str == "YES")
        {
            vector<int> v(max,0);  
        vector<int>::iterator it;
        it = set_intersection(yes.begin(),yes.end(), tmp.begin(), tmp.end(), v.begin());
    v.resize(it-v.begin());
 
    
            yes.clear();
            for(it = v.begin();it != v.end();it++)
                yes.insert(*it);
            tmp.clear();
        }
        if (str =="NO"){
             
            vector<int> v(max,0);  
        vector<int>::iterator it;
        it = set_union(no.begin(),no.end(), tmp.begin(), tmp.end(), v.begin());
    v.resize(it-v.begin());
 
    
            no.clear();
            for(it = v.begin();it != v.end();it++)
                no.insert(*it);
            tmp.clear();
        }
 
        
    
    }
 vector<int> v(max,0);  
  std::vector<int>::iterator it;
    it = set_difference (yes.begin(),yes.end(), no.begin(), no.end(), v.begin());
    v.resize(it-v.begin());
    for(it = v.begin();it != v.end();it++)
        cout<<*it<<" ";
        cout<<endl;
 
    return 0;
}

#include <iostream>
#include <set>
#include <string>
#include <sstream>
#include <algorithm>
#include <vector>
#include <iterator>
using namespace std;
 
int main(int argc, char* argv[])
{
    set<int> yes, no;
    int max;
    string str;
    cin >> max;
 
    while (str != "HELP")
    {
        getline(cin, str);
        stringstream ss;
        ss.str(str);
        set<int> input;
        int word;
        while (ss >> word)
        {
            input.insert(word);
        }
        if (str == "YES")
        {
            if(yes.empty())
            {
                // не разу не был YES
                yes = input;
                if(!no.empty())
                {
                    //вычитаем множество no из yes
                    std::set<int> result;
                    set_difference(yes.begin(), yes.end(), no.begin(), no.end(),
                            inserter(result, result.end()));
                    yes = result;
                    no.clear();
                }
            }
            else
            {
                // второй и более YES
                std::set<int> result;
                set_intersection(yes.begin(), yes.end(), input.begin(), input.end(),
                        inserter(result, result.end()));
                yes = result;
            }
        }
        if (str == "NO")
        {
            if(yes.empty())
            {
                // не было ни одного YES
                // накапливаем NO
                no.insert (input.begin (), input.end());
            }
            else
            {
                //вычитаем входное множество из yes
                std::set<int> result;
                set_difference(yes.begin(), yes.end(), input.begin(), input.end(),
                        inserter(result, result.end()));
                yes = result;
            }
        }
    }
 
    if(yes.empty() && !no.empty())
    {
        // не было ни одного YES
        // выводим разность [1..n] и no
        typedef set<int>::const_iterator Iterator;
        Iterator noItr(no.begin());
        Iterator endItr(no.end());
        for(int i = 1; i < max; ++i)
        {
            if(noItr != endItr)
            {
                if(i<*noItr)
                {
                    std::cout<<i;
                }
                else
                {
                    // i есть в NO
                    // пропускаем
                    ++noItr;
                }
 
            }
            else
            {
                cout<<i<<" ";
            }
 
        }
    }
    if(!yes.empty() && no.empty())
    {
        // ответ в yes.
        //...
    }
    if(yes.empty() && no.empty())
    {
        // вырожденный случай
        // не было ни YES, ни NO, а сразу HELP
        for(int i = 1; i< max; ++i)
        {
            cout<<i<<" ";
        }
 
    }
    return 0;
}