Submission #1268572

#	Time	Username	Problem	Language	Result	Execution time	Memory
1268572		ducdev	Genetics (BOI18_genetics)	C++17	74 / 100	2100 ms	91144 KiB

genetics

// Author: 4uckd3v - Nguyen Cao Duc
#include <bits/stdc++.h>
using namespace std;

#define all(x) (x).begin(), (x).end()

typedef long long ll;

const int MAX_N = 4100;
const int MOD = 1e9 + 7;

int m, n, k;
char a[MAX_N + 5][MAX_N + 5];

namespace SUBTASK_1 {
    const int LIM = 100;

    int diff(int i, int j) {
        int cnt = 0;
        for (int k = 1; k <= n; k++)
            if (a[i][k] != a[j][k]) cnt++;
        return cnt;
    };

    void Solve() {
        for (int i = 1; i <= m; i++) {
            bool real = true;
            for (int j = 1; j <= m; j++) {
                if (i == j) continue;
                if (diff(i, j) != k) {
                    real = false;
                    break;
                };
            };
            if (real) {
                cout << i << '\n';
                break;
            };
        };
    };
};  // namespace SUBTASK_1

namespace SUBTASK_23 {
    const int M = MAX_N;
    const int N = MAX_N;

    int canReal[M + 5];
    int cache[M + 5][M + 5];
    bitset<N> bs[M];

    bool checkSubtask() {
        for (int i = 1; i <= m; i++)
            for (int j = 1; j <= n; j++)
                if (a[i][j] != 'A' && a[i][j] != 'C') return false;
        return true;
    };

    void Solve() {
        for (int i = 1; i <= m; i++) {
            for (int j = 1; j <= n; j++) {
                bs[i - 1].set(j - 1, a[i][j] == 'A');
            };
        };

        memset(cache, -1, sizeof cache);
        for (int i = 0; i <= m; i++) canReal[i] = true;

        for (int i = 0; i < m; i++) {
            if (!canReal[i]) continue;

            for (int j = 0; j < m; j++) {
                if (i == j) continue;

                if (cache[i][j] == -1) cache[i][j] = cache[j][i] = (bs[i] ^ bs[j]).count();

                if (cache[i][j] != k) {
                    canReal[i] = canReal[j] = false;
                    break;
                };
            };

            if (canReal[i]) {
                cout << i + 1 << '\n';
                break;
            };
        };
    };

};  // namespace SUBTASK_23

namespace SUBTASK_4 {
    const int M = MAX_N;
    const int N = MAX_N;

    int id[256], perm[M + 5];
    int cache[M + 5][M + 5];
    bool canReal[M + 5];
    bitset<N> bs[M][4];

    void Solve() {
        id['A'] = 0;
        id['T'] = 1;
        id['G'] = 2;
        id['C'] = 3;

        memset(cache, -1, sizeof cache);
        for (int i = 0; i <= m; i++) canReal[i] = true;
        for (int i = 1; i <= m; i++) {
            for (int j = 1; j <= n; j++) {
                bs[i - 1][id[a[i][j]]].set(j - 1);
            };
        };

        for (int i = 0; i < m; i++) perm[i] = i;
        random_shuffle(perm, perm + m);

        for (int i = 0; i < m; i++) {
            if (!canReal[perm[i]]) continue;

            for (int j = 0; j < m; j++) {
                if (i == j) continue;

                if (cache[perm[i]][perm[j]] == -1) {
                    int numCorrect = 0;
                    for (int type = 0; type < 4; type++) {
                        numCorrect += (bs[perm[i]][type] & bs[perm[j]][type]).count();
                        if (numCorrect > n - k) break;
                    };
                    cache[perm[i]][perm[j]] = cache[perm[i]][perm[j]] = numCorrect;
                };

                if (cache[perm[i]][perm[j]] != n - k) {
                    canReal[perm[i]] = canReal[perm[j]] = false;
                    break;
                };
            };

            if (canReal[perm[i]]) {
                cout << perm[i] + 1 << '\n';
                break;
            };
        };
    };
};  // namespace SUBTASK_4

int main() {
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    if (fopen("MAIN.INP", "r")) {
        freopen("MAIN.INP", "r", stdin);
        freopen("MAIN.OUT", "w", stdout);
    };

    cin >> m >> n >> k;
    for (int i = 1; i <= m; i++) {
        for (int j = 1; j <= n; j++) {
            cin >> a[i][j];
        };
    };
  
    if (m <= 100 && n <= 100)
      return SUBTASK_1::Solve(), 0;
    if (SUBTASK_23::checkSubtask())
      return SUBTASK_23::Solve(), 0;
    SUBTASK_4::Solve();
};

Compilation message (stderr)

genetics.cpp: In function 'void SUBTASK_4::Solve()':
genetics.cpp:115:23: warning: 'void std::random_shuffle(_RAIter, _RAIter) [with _RAIter = int*]' is deprecated: use 'std::shuffle' instead [-Wdeprecated-declarations]
  115 |         random_shuffle(perm, perm + m);
      |         ~~~~~~~~~~~~~~^~~~~~~~~~~~~~~~
In file included from /usr/include/c++/13/algorithm:61,
                 from /usr/include/x86_64-linux-gnu/c++/13/bits/stdc++.h:51,
                 from genetics.cpp:2:
/usr/include/c++/13/bits/stl_algo.h:4581:5: note: declared here
 4581 |     random_shuffle(_RandomAccessIterator __first, _RandomAccessIterator __last)
      |     ^~~~~~~~~~~~~~
genetics.cpp: In function 'int main()':
genetics.cpp:150:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
  150 |         freopen("MAIN.INP", "r", stdin);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~
genetics.cpp:151:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
  151 |         freopen("MAIN.OUT", "w", stdout);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~

• Subtask 1: 27 / 27
• Subtask 2: 19 / 19
• Subtask 3: 28 / 28
• Subtask 4: 0 / 26