cin 没有接受 while 循环中的最后一个输入

Question

在以下代码段中，while 循环内的 cin 嵌套在 main 函数中的 while 循环内，在最后一次输入期间（即，当 @987654326 @) 并改为报告分段错误。

我通过使用print 语句（在下面的代码中注释）来解决这个问题，一个在内部while 循环的开头，一个在cin 语句之后（查找读取的值v 和 x 减 1) 和 while 循环外。第三个cout 语句没有执行，由此我断定分段错误在while 循环内。

此外，在内部while 循环的最后一次迭代中，cin 语句之后的cout 语句（用于打印v 和x 的值）也没有执行。因此，我认为内部 while 循环中的 cin 在最后一次迭代中不起作用。

我尝试了很多搜索并输入了cin.clear() 和cin.ignore(numeric_limits<streamsize>::max(), '\n'); 语句，但仍然没有用！

为什么会出现这样的问题？

#include <bits/stdc++.h>

using namespace std;
vector <vector<int>> adj;
vector<unordered_set <int>> leaf;

int process(int node, int *pans, int k){
    *pans = *pans + (leaf[node].size() / k) * k;
    int temp = (leaf[node].size() / k) * k;
    if(temp == 0) return 0;
    for(auto it : leaf[node]){
        if(temp == 0) break;
        leaf[node].erase(it);
        auto j = find(adj[node].begin(), adj[node].end(), it);
        adj[node].erase(j);
        temp --;
    }
    if(adj[node].size() == 1){
        leaf[adj[node][0]].insert(node);
        process(adj[node][0], pans, k);
    }
    return 0;
}

int main(){
    int t, v, x, n, k;
    cin>>t;
    cin.clear();
    cin.ignore(numeric_limits<streamsize>::max(), '\n');
    while(t--){
        int ans = 0;
        cin>>n>>k;
        cin.clear();
        cin.ignore(numeric_limits<streamsize>::max(), '\n');
        adj.resize(n);
        int h = 1;
        while(h < n){
            // cout<<"r";
            cin>>v>>x;
            cin.clear();
            cin.ignore(numeric_limits<streamsize>::max(), '\n');
            v --; x --;
            cout<<v<<" "<<x<<"\n";
            adj[v].push_back(x);
            adj[x].push_back(v);
            h++;
        }
        // cout<<"out";
        leaf.resize(n);
        for(int i = 0; i < n; i ++){
            if(adj[i].size() == 1) leaf[adj[i][0]].insert(i);
        }
        for(int i = 0; i < n; i ++){
            if(leaf[i].size() >= k) process(i, &ans, k);
        }
        cout<<ans<<"\n";
        adj.clear();
        leaf.clear();
    }
}

以下是输入示例：

1
8 3
1 2
1 5
7 6
6 8
3 1
6 4
6 1

Answer 1

错误在这里

for(auto it : leaf[node]){
    if(temp == 0) break;
    leaf[node].erase(it);
    auto j = find(adj[node].begin(), adj[node].end(), it);
    adj[node].erase(j);
    temp --;
}

通过修改 unordered_set leaf[node].erase(it);，您将使基于范围的 for 循环使用的隐式迭代器无效。用一个显式迭代器重写你的循环，这样你就可以解决迭代器失效的问题。像这样

for (auto i = leaf[node].begin(); i != leaf[node].end(); ) {
    if(temp == 0) break;
    auto it = *i;
    i = leaf[node].erase(i);
    auto j = find(adj[node].begin(), adj[node].end(), it);
    adj[node].erase(j);
    temp --;
}

通过这个循环，您的代码将运行完成（至少对我而言）。我不知道输出是否正确。