简体   繁体   中英

Nested range-based for-loops

I have the following code using range-based for -loops (C++11):

vector<atom> protein;
...
for(atom &atom1 : protein) {
    ...
    for(atom &atom2 : protein) {
        if(&atom1 != &atom2) {
                ...
        }
    }
}

Is there a better/cleaner/faster way to write this nested loops? Isn't there a way to include the if condition in the second loop?

Similar to ronag's answer is a more generic version:

template<typename C, typename Op>
void each_unique_pair(C& container, Op fun)
{
    for(auto it = container.begin(); it != container.end() - 1; ++it)
    {
        for(auto it2 = std::next(it); it2 != container.end(); ++it2)
        {
            fun(*it, *it2);
            fun(*it2, *it);
        }
    }
}

UPDATE

template<typename C, typename O1, typename O2>
void each_value_and_pair(C& container, O1 val_fun, O2 pair_fun)
{
    auto it = std::begin(container);
    auto end = std::end(container);
    if(it == end)
        return;

    for(; it != std::prev(end); ++it)
    {
        val_fun(*it);
        for(auto it2 = std::next(it); it2 != end; ++it2)
        {
            pair_fun(*it2, *it);
            pair_fun(*it, *it2);
        }
    }
}

Which is used like this:

main()
{
    std::vector<char> values;
    // populate values
    // ....
    each_value_and_pair(values, 
        [](char c1) { std::cout << "value: " << c1 << std::endl;}, 
        [](char c1, char c2){std::cout << "pair: " << c1 << "-" << c2 << std::endl;});
}

Sad but true. How about normal loops with iterators and auto keyword?

I think this might be what you are looking for:

for(auto it1 = std::begin(protein1); it1 != std::end(protein); ++it1)
{
    for(auto it2 = std::next(it1); it2 != std::end(protein); ++it2)
    {
          auto& atom1 = *it1;
          auto& atom2 = *it2;

           // ...
    }
}

you're method is just fine. if you want to save the if statement you can

vector<atom> protein;
int i, j;
...
for(i = 0; i < protein.size() : i++) {
    atom &atom1 = protein.at(i);
    for(j = i+1; j < protein.size() ; j++) {
        atom &atom2 = protein.at(j);
                    // Do something
                    // Swap the atom2 and atom1
                    // Do something again
    }
}

The technical post webpages of this site follow the CC BY-SA 4.0 protocol. If you need to reprint, please indicate the site URL or the original address.Any question please contact:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM