PAT 1076 Forwards on Weibo

#include <cstdio>
#include <cstdlib>

#include <vector>
#include <queue>
#include <unordered_set>

using namespace std;

class User {
public:
    vector<int> fellowers;
    bool visited;
    User(): visited(false){}
};

int query(int idx, vector<User>& users, int L) {
    int level = L;
    unordered_set<int> walked;
    queue<int> level_queue;

    walked.insert(idx);
    level_queue.push(idx);

    int last_size = level_queue.size();

    while (!level_queue.empty() && level > ) {
        while(last_size-- > ) {
            int uid = level_queue.front();

            User& user = users[uid];
            level_queue.pop();
            int len = user.fellowers.size();
            for (int i=; i<len; i++) {
                int u = user.fellowers[i];
                // already visited
                if (walked.count(u)) continue;
                // not visited
                //printf("visit %d\n", u);
                walked.insert(u);
                level_queue.push(u);
            }
        }

        last_size = level_queue.size();
        level--;
    }

    return walked.size() - ;
}

int main() {
    int N = , L = ;
    scanf("%d%d", &N, &L);

    vector<User> users(N + );

    for (int i=; i<=N; i++) {
        int n = ;
        scanf("%d", &n);
        for (int j=; j<n; j++) {
            int uidx = ;
            scanf("%d", &uidx);
            users[uidx].fellowers.push_back(i);
        }
    }

    int K = ;
    scanf("%d", &K);
    for (int i=; i<K; i++) {
        int uidx = ;
        scanf("%d", &uidx);
        int cnt = query(uidx, users, L);
        printf("%d\n", cnt);
    }
    return ;
}

最后一个case用时有1800ms，不过还是在范围之内，应该有可以再优化的地方