牛客-栈、队列和stl-2022.10.14

Keep In Line

用数组模拟队列，用unordered_map记录每个名字的同学的位置，之后判断当前队列的第一个名字和出队的人的名字是否相同，如果相同则将头指针下标移动到下一个，符合排队规则的人数+1，否则更新当前位置之后的所有人的名字的位置（更新unordered_map）

#include <bits/stdc++.h>
using namespace std;
string Line[100010];
int main() {
    int t;
    cin >> t;
    while(t--) {
        int n;
        cin >> n;
        unordered_map<string, int> p;
        int res = 0, start = 0, end = 0;
        for(int i = 0; i < n; i++) {
            string Opt, Name;
            cin >> Opt >> Name;
            if(Opt == "in") {
                Line[end] = Name;
                p[Name] = end;
                end += 1;
            } else if(Opt == "out") {
                if(Line[start] == Name) {
                    res += 1;
                    start += 1;
                } else {
                    if(p[Name] == end - 1)
                        p[Name] = -1;
                    else {
                        for(int j = p[Name] + 1; j < end; j++) {
                            Line[j - 1] = Line[j];
                            p[Line[j]] -= 1;
                        }
                    }
                    end -= 1;
                }
            }
        }
        cout << res << endl;
    }
    return 0;
}

Number

先记录小于最大数值的所有的质数，然后对于这些质数进行2次方，3次方，4次方求和的三重循环遍历，考虑到最大的质数数据范围小于一万，数据范围为10^3，所以直接遍历记录相加求和的结果即可。注意去重！

#include <bits/stdc++.h>
using namespace std;
const int NUM = 10000;
bool vis[10010] = {false};
vector<int> prime;
void init() {
    for(int i = 2; i <= sqrt(NUM + 0.5); i++) {
        if(!vis[i]) {
            for(int j = i * i; j <= NUM; j += i)
                vis[j] = true;
        }
    }
    for(int i = 2; i <= NUM; i++) {
        if(!vis[i])
            prime.emplace_back(i);
    }
}
int main() {
    init();
    int n;
    cin >> n;
    unordered_set<long long> p;
    int size = prime.size();
    //cout << size << endl;
    for(int i = 0; i < size; i++) {
        long long sum1 = prime[i] * prime[i];
        if(sum1 > n)
            break;
        for(int j = 0; j < size; j++) {
            long long sum2 = sum1 + prime[j] * prime[j] * prime[j];
            if(sum2 > n)
                break;
            for(int k = 0; k < size; k++) {
                long long sum3 = sum2 + prime[k] * prime[k] * prime[k] * prime[k];
                if(sum3 > n)
                    break;
                p.insert(sum3);           
            }
        }
    }
    cout << p.size() << endl;
    return 0;
}