QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#770954#8212. Football in Osijektassei903WA 1ms3524kbC++233.2kb2024-11-22 04:39:062024-11-22 04:39:06

Judging History

你现在查看的是最新测评结果

  • [2024-11-22 04:39:06]
  • 评测
  • 测评结果:WA
  • 用时:1ms
  • 内存:3524kb
  • [2024-11-22 04:39:06]
  • 提交

answer

#include <bits/stdc++.h>

#define rep1(n) for(int i = 0; i < (int)n; i++)
#define rep2(i, n) for(int i = 0; i < (int)n; i++)
#define rep3(i, l, r) for(int i = (int)l; i < (int)r; i++)
#define overloadrep(a, b, c, x, ...) x
#define rep(...) overloadrep(__VA_ARGS__, rep3, rep2, rep1)(__VA_ARGS__)
#define all(v) v.begin(), v.end()
#define sz(v) (int)v.size()
using namespace std;

using ll = long long;
using vi = vector<int>;


void solve() {
    int n;cin >> n;
    vector<int> p(n);for (auto &x: p) cin >> x, x--;
    vector<int> deg(n, 0);
    vector<vi> rev(n);
    rep(i, n)deg[p[i]]++;
    queue<int> q;
    rep(i, n)if (deg[i] == 0) q.push(i);
    vector<int> dp(n, 0);
    vector<int> order;
    while (!q.empty()) {
        int x = q.front();q.pop();
        order.emplace_back(x);
        rev[p[x]].emplace_back(x);
        dp[x] = -1;
        deg[p[x]]--;
        if (deg[p[x]] == 0)q.push(p[x]);
    }

    rep(i, n) {
        if (dp[i] != 0) continue;
        int cnt = 0, x = i;
        do {
            cnt++;
            x = p[x];
            
        } while (x != i);
        
        do {
            dp[x] = cnt;
        }while (x != i);
    }
    vector<bool> flag(n+1);
    for (int i : order | views :: reverse) {
        dp[i] = dp[p[i]] + 1;
        flag[dp[i]] = true;
    }
    for (int i : order | views :: reverse) {
        dp[i] = -1;
    }
    
    vector<int> res;
    auto dfs = [&](auto self, int v) -> int {
        // cout << v + 1 << endl;
        vector<int> f;
        for (int u: rev[v]) {
            f.emplace_back(self(self, u) + 1);
            if (f.back() > f.front()) swap(f.back(), f.front());
        }
        // cout << v + 1 << " :";
        for (int x : f | views::drop(1)) res.emplace_back(x);
        // for (int x : f) cout << x << " ";
        // cout << endl;
        
        if (sz(f)) return f.front();
        return 0;
    };

    rep(i, n) if (dp[i] != -1) {
        // cout << i + 1 << " " << dp[i] << endl;
        vector<int> f;
        int v = i;
        int y = dp[i];
        do {
            f.emplace_back(dfs(dfs, v));
            if (f.back() > f.front()) swap(f.back(), f.front());
            flag[dp[v]] = true;
            dp[v] = -1;
            v = p[v];
        } while (v != i);
        for (int x : f | views::drop(1)) if (x) res.emplace_back(x);
        // for (int x : f) cout << x << " ";
        // cout << endl;
        res.emplace_back(f.front() + y);
    }

    // rep(i, n) {
    //     cout << i + 1 << " : ";
    //     for (int x : rev[i])cout << x+1 << " ";
    //     cout << endl;
    // }
    // for (int x : res) cout << x <<  " ";
    // cout << endl;
    sort(all(res));
    reverse(all(res));

    int x = 0;
    rep(i, sz(res)) {
        rep(j, res[i]) {
            dp[x + j] = i;
        }
        x += res[i];
    }

    rep(i, n) {
        if (i) cout << " ";
        if (flag[i+1]) cout << 0;
        else cout << dp[i] + 1;
    }
    cout << endl;

}

int main() {

    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    int T = 1;
    // int T;cin >> T;
    while(T--) {
        solve();
    }
    
} 

Details

Tip: Click on the bar to expand more detailed information

Test #1:

score: 0
Wrong Answer
time: 1ms
memory: 3524kb

input:

5
2 3 1 3 5

output:

0 1 0 0 2

result:

wrong answer 5th numbers differ - expected: '1', found: '2'