美团2021校招笔试-编程题(通用编程试题,第9场)

T3

环形的最大连续子数和
分成两种情况：

1. 这个最大子数组没有跨数组
2. 这个最大子数组跨了数组

分别对应下面两种情况：

1. 先求（不跨数组）连续最大子数组和
2. 再求（不跨数组）连续最小子数组和，sum-最小子数组和=最大子数组和（跨数组）
   这两个值求一个max就是结果

我一个憨批还写滑动窗口

#include<bits/stdc++.h>

using namespace std;

const int N = 100010;
int w[N];

int main() {
    int T;
    cin >> T;

    while (T--) {
        int n, res = 0;
        cin >> n;
        for (int i = 0; i < n; i++)cin >> w[i];
        queue<int> q;
        int cur = 0;
        int sum = 0;
        for (int i = 0; i < n; i++) {
            sum += w[i];
            cur = max(0, cur + w[i]);
            res = max(res, cur);
        }

        cur = 0;
        int mn = 0;
        for (int i = 0; i < n; i++) {
            cur = min(0, cur + w[i]);
            mn = min(mn, cur);
        }
        res = max(res, sum - mn);
        cout << res << endl;
    }
}

T4

思路是对的，就是没写对
因为看见C的范围很大，其实color的种类最多是n个，因此写了两个map做了一个映射。
但是不知道为啥wa了，也懒的调了。

找了一个别人的代码改了改
这个人用了multiset很灵性，用multiset的count去计算颜色数量。
我没咋用过multiset，不太熟悉。

#include <bits/stdc++.h>

using namespace std;
typedef long long ll;
vector<int> e[5100];
int c[5100], res[5100];
set<int> st;
multiset<int> mt[5100];


void dfs(int root, int fa) {
    for (auto i:e[root]) {
        if (i == fa)continue;
        dfs(i, root), mt[root].insert(mt[i].begin(), mt[i].end());
    }
    mt[root].insert(c[root]);
    int mx = 0, cat;
    for (auto i:st) {
        int num = mt[root].count(i);
        if (num > mx)
            mx = num, cat = i;
    }
    res[root] = cat;
}

int main() {
    int n, x, y, q, m;
    cin >> n;
    for (int i = 1; i <= n - 1; ++i) {
        cin >> x >> y;
        e[x].push_back(y), e[y].push_back(x);
    }

    for (int i = 1; i <= n; ++i) {
        cin >> c[i];
        st.insert(c[i]);
    }

    dfs(1, -1);
    cin >> q;
    for (int i = 0; i < q; ++i) {
        int t;
        cin >> t;
        printf("%d\n", res[t]);
    }
    return 0;
}