AtCoder Beginner Contest 252（D - Distinct Trio & E - Road Reduction & F - Bread）

比赛链接

AtCoder Beginner Contest 252

D - Distinct Trio

原题链接

D - Distinct Trio

题面

思路

• 给你一个序列，求三元组得个数，因为要找三个数字，直接求的话显然复杂度不行，我们可以把中间数字$A_j$固定，求有多少个数字比$A_j$小，有多少个数字比$A_j大，就是可以统计每一个数字出现的次数，然后求前缀和即可。

代码

const int N = 2e5+100;
void solve(){
    int n; cin >> n;
    vector<int> a(n), s(N);
    for(int i = 0; i < n; i++) {
        cin >> a[i];
        s[a[i]] ++;
    }

    sort(all(a));
    a.erase(unique(all(a)), a.end());
    for(int i = 1; i < N; i++) s[i] += s[i - 1];

    ll ans = 0;
    for(int i = 0; i < a.size(); i++) {
        ans += 1ll * s[a[i] - 1] * (n - s[a[i]]) * (s[a[i]] - s[a[i] - 1]);
        // 比Aj小的 * 比Aj大的 * Aj个数
    }

    cout << ans << '\n';
}

E - Road Reduction

原题链接

E - Road Reduction

题面

思路

• n个点m条边，从1号点开始到第i个点的最短距离为$d_i$，求$d_2+d_3+...+d_n$最小值，最短路生成树。首先题目保证图是联通的，从1好点开始跑一个dijksra求出从1号点到每个点的最短距离，然后将连接这些点的边输出。

代码

const int N = 2e5+100;
int h[N], id[N * 2], e[N * 2], ne[N * 2], w[N * 2];
ll dist[N], idx;
void add(int a, int b, int c, int d) {
    e[idx] = b, id[idx] = d, w[idx] = c, ne[idx] = h[a], h[a] = idx ++;
}
bool vis[N];
void solve(){
    memset(h, -1, sizeof h);
    int n, m; cin >> n >> m;
    for(int i = 1; i <= m; i++) {
        int a, b, c; cin >> a >> b >> c;
        add(a, b, c, i); add(b, a, c, i);
    }
 
    function<void()> dijkstra = [&](){
        memset(dist, 0x3f, sizeof dist);
        dist[1] = 0;
        priority_queue<pair<ll, int>> pq;
        pq.push({0, 1});
 
        while(pq.size()) {
            auto t = pq.top().second; pq.pop();
 
            if(vis[t]) continue;
            vis[t] = 1;
 
            for(int i = h[t]; ~i; i = ne[i]) {
                int j = e[i];
                if(dist[j] > dist[t] + w[i]) {
                    dist[j] = dist[t] + w[i];
                    pq.push({-dist[j], j});
                }
            }
        }
    };
 
    dijkstra();
 
    for(int u = 2; u <= n; u++) {
        for(int i = h[u]; ~i; i = ne[i]) {
            int j = e[i];
            if(dist[u] == dist[j] + w[i]) {
                cout << id[i] << ' ';
                break;
            }
        }
    }
}

F - Bread

原题链接

F - Bread

题面

思路

• 每次分割完面包后代价是分割前的长度，想到合并果子那题类似，详细看下面代码。

代码

void solve(){
    ll n, l; cin >> n >> l;
    priority_queue<ll, vector<ll>, greater<ll>> pq;
    for(int i = 0; i < n; i++) {
        int u; cin >> u;
        pq.push(u);
        l -= u;
    }
    if(l) pq.push(l);
    ll ans = 0;
    while(pq.size() > 1) {
        auto l = pq.top(); pq.pop();
        auto r = pq.top(); pq.pop();
        ans += l + r;
        pq.push(l + r);
    }
    cout << ans << '\n';
}