美團2021校招筆試-編程題(通用編程試題,第9場)

T3

環(huán)形的最大連續(xù)子數(shù)和
分成兩種情況：

1. 這個最大子數(shù)組沒有跨數(shù)組
2. 這個最大子數(shù)組跨了數(shù)組

分別對應下面兩種情況：

1. 先求（不跨數(shù)組）連續(xù)最大子數(shù)組和
2. 再求（不跨數(shù)組）連續(xù)最小子數(shù)組和，sum-最小子數(shù)組和=最大子數(shù)組和（跨數(shù)組）
   這兩個值求一個max就是結(jié)果

我一個憨批還寫滑動窗口

#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了，也懶的調(diào)了。

找了一個別人的代碼改了改
這個人用了multiset很靈性，用multiset的count去計算顏色數(shù)量。
我沒咋用過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;
}