oj.uz

답안 #944603

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
944603 2024-03-13T01:38:29 Z Nhoksocqt1 가장 긴 여행 (IOI23_longesttrip) C++17
5 / 100
 1000 ms 856 KB 
#ifndef Nhoksocqt1
    #include "longesttrip.h"
#endif // Nhoksocqt1
#include<bits/stdc++.h>
using namespace std;

#define inf 0x3f3f3f3f
#define sz(x) int((x).size())
#define fi first
#define se second
typedef long long ll;
typedef pair<int, int> ii;

mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());
int Random(int l, int r) {
    return uniform_int_distribution<int>(l, r)(rng);
}

const int MAXN = 260;

#ifdef Nhoksocqt1

bitset<MAXN> adjp[MAXN], idA, idB, bsA, bsB;
struct Jury {
    int n, d;

    void init(int _n, int _d) {
        n = _n, d = _d;
        int u, v;
        bool check(0);
        while(cin >> u >> v) {
            adjp[u][v] = adjp[v][u] = 1;
            check = 1;
        }

        if(!check) {
            for (int i = 0; i < n; ++i) {
                for (int j = i + 1; j < n; ++j) {
                    for (int k = j + 1; k < n; ++k) {
                        int cnt = adjp[i][j] + adjp[i][k] + adjp[j][k];
                        if(cnt < d) {
                            vector<ii> p;
                            p.push_back(ii(i, j));
                            p.push_back(ii(i, k));
                            p.push_back(ii(j, k));

                            while(cnt < d) {
                                int r = Random(0, 2);
                                if(adjp[p[r].fi][p[r].se])
                                    continue;

                                ++cnt;
                                adjp[p[r].fi][p[r].se] = adjp[p[r].se][p[r].fi] = 1;
                            }
                        }
                    }
                }
            }
        }
    }

    bool are_connected(vector<int> &A, vector<int> &B) {
        bsA = bsB = idA = idB = 0;
        for (int it = 0; it < sz(A); ++it) {
            bsA |= adjp[A[it]];
            idA[A[it]] = 1;
        }

        for (int it = 0; it < sz(B); ++it) {
            bsB |= adjp[B[it]];
            idB[B[it]] = 1;
        }

        return ((bsA & idB).count() > 0 || (bsB & idA).count() > 0);
    }

} jury;

#endif // Nhoksocqt1

int numNode;
bool adj[MAXN][MAXN], dx[MAXN];

bool check_are_connected(vector<int> &A, vector<int> &B) {
    #ifdef Nhoksocqt1
        return jury.are_connected(A, B);
    #else
        return are_connected(A, B);
    #endif // Nhoksocqt1
}

int evaluation(int u) {
    vector<int> tmp;
    while(1) {
        dx[u] = 1;
        tmp.push_back(u);

        int maxLen(0), v(-1);
        for (int i = 0; i < numNode; ++i) {
            if(!dx[i] && adj[u][i]) {
                v = i;
                break;
            }
        }

        if(v < 0)
            break;

        u = v;
    }

    int res = sz(tmp);
    while(sz(tmp)) {
        dx[tmp.back()] = 0;
        tmp.pop_back();
    }

    return res;
}

int evaluation2(int u) {
    vector<int> tmp;
    while(1) {
        dx[u] = 1;
        tmp.push_back(u);

        int maxLen(0), v(-1);
        for (int i = 0; i < numNode; ++i) {
            if(!dx[i] && adj[u][i]) {
                int len = evaluation(i);
                if(maxLen < len) {
                    maxLen = len;
                    v = i;
                }
            }
        }

        if(v < 0)
            break;

        u = v;
    }

    int res = sz(tmp);
    while(sz(tmp)) {
        dx[tmp.back()] = 0;
        tmp.pop_back();
    }

    return res;
}

vector<int> longest_trip(int N, int D) {
    numNode = N;
    if(D == 3) {
        vector<int> ans;
        for (int i = 0; i < numNode; ++i)
            ans.push_back(i);

        return ans;
    }

    for (int u = 0; u < numNode; ++u) {
        for (int v = u + 1; v < numNode; ++v) {
            vector<int> A, B;
            A.push_back(u);
            B.push_back(v);
            adj[u][v] = adj[v][u] = check_are_connected(A, B);
        }
    }

    for (int i = 0; i < numNode; ++i)
        dx[i] = 0;

    vector<int> ans;
    int len(0), u(-1);
    for (int i = 0; i < numNode; ++i) {
        int len = evaluation2(i);
        if(len >= numNode / 2) {
            u = i;
            break;
        }
    }

    while(1) {
        dx[u] = 1;
        ans.push_back(u);

        int maxLen(0), v(-1);
        for (int i = 0; i < numNode; ++i) {
            if(!dx[i] && adj[u][i]) {
                int len = sz(ans) + evaluation(i);
                if(len + sz(ans) >= numNode / 2) {
                    v = i;
                    break;
                }
            }
        }

        if(v < 0)
            break;

        u = v;
    }

    return ans;
}

#ifdef Nhoksocqt1

int main(void) {
    ios_base::sync_with_stdio(0), cin.tie(0), cout.tie(0);

    #define TASK "longesttrip"
    if(fopen(TASK".inp", "r")) {
        freopen(TASK".inp", "r", stdin);
        freopen(TASK".out", "w", stdout);
    }

    int n, d;
    cin >> n >> d;
    jury.init(n, d);

    vector<int> path = longest_trip(n, d);
    cout << "ANSWER: " << sz(path) << ": ";
    for (int it = 0; it < sz(path); ++it)
        cout << path[it] << ' '; cout << '\n';

    return 0;
}

#endif // Nhoksocqt1

Compilation message

longesttrip.cpp: In function 'int evaluation(int)':
longesttrip.cpp:98:13: warning: unused variable 'maxLen' [-Wunused-variable]
   98 |         int maxLen(0), v(-1);
      |             ^~~~~~
longesttrip.cpp: In function 'std::vector<int> longest_trip(int, int)':
longesttrip.cpp:189:13: warning: unused variable 'maxLen' [-Wunused-variable]
  189 |         int maxLen(0), v(-1);
      |             ^~~~~~
longesttrip.cpp:176:9: warning: unused variable 'len' [-Wunused-variable]
  176 |     int len(0), u(-1);
      |         ^~~

Subtask #0
0.0 / 0.0

# 결과 실행 시간 메모리 Grader output
1 Correct 0 ms 344 KB Output is correct
2 Correct 797 ms 484 KB Output is correct

Subtask #1
5.0 / 5.0

# 결과 실행 시간 메모리 Grader output
1 Correct 2 ms 344 KB Output is correct
2 Correct 1 ms 344 KB Output is correct
3 Correct 1 ms 344 KB Output is correct
4 Correct 1 ms 344 KB Output is correct
5 Correct 1 ms 344 KB Output is correct

Subtask #2
0.0 / 10.0

# 결과 실행 시간 메모리 Grader output
1 Correct 7 ms 344 KB Output is correct
2 Correct 24 ms 344 KB Output is correct
3 Correct 152 ms 344 KB Output is correct
4 Correct 707 ms 344 KB Output is correct
5 Execution timed out 3056 ms 856 KB Time limit exceeded
6 Halted 0 ms 0 KB -

Subtask #3
0.0 / 25.0

# 결과 실행 시간 메모리 Grader output
1 Correct 8 ms 344 KB Output is correct
2 Correct 29 ms 344 KB Output is correct
3 Correct 160 ms 344 KB Output is correct
4 Correct 713 ms 344 KB Output is correct
5 Execution timed out 3026 ms 480 KB Time limit exceeded
6 Halted 0 ms 0 KB -

Subtask #4
0.0 / 60.0

# 결과 실행 시간 메모리 Grader output
1 Correct 7 ms 344 KB Output is correct
2 Correct 26 ms 344 KB Output is correct
3 Partially correct 162 ms 344 KB Output is partially correct
4 Partially correct 758 ms 344 KB Output is partially correct
5 Execution timed out 3099 ms 480 KB Time limit exceeded
6 Halted 0 ms 0 KB -