Submission #1162841

#	Time	Username	Problem	Language	Result	Execution time	Memory
1162841		belgianbot	Genetics (BOI18_genetics)	C++20	0 / 100	2097 ms	49040 KiB

genetics

/*
    AUTHOR : BELGIANBOT
*/

#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
using namespace std;
using namespace __gnu_pbds;

template<typename T> void read(T& x) { cin >> x;}
template<typename T1, typename T2> void read(pair<T1, T2>& p) { read(p.first), read(p.second);}
template<typename T> void read(vector<T>& v) { for (auto& x : v) read(x); }
template<typename T1, typename T2> void read(T1& x, T2& y) { read(x), read(y); }
template<typename T1, typename T2, typename T3> void read(T1& x, T2& y, T3& z) { read(x), read(y), read(z); }
template<typename T1, typename T2, typename T3, typename T4> void read(T1& x, T2& y, T3& z, T4& zz) { read(x), read(y), read(z), read(zz); }
template<typename T> void print(vector<T>& v) { for (auto& x : v) cout << x << ' '; cout << endl; }

#define quit(x) return (void)(cout << x << endl)
#define pb push_back
#define fi first
#define se second
#define ordered_set tree<pair<int,int>, null_type, less<pair<int,int>>, rb_tree_tag, tree_order_statistics_node_update>
#define all(x) (x).begin(), (x).end()
#define rall(x) (x).rbegin(), (x).rend()
#define mp make_pair
#define sz(x) (int)(x.size())
#define YES cout<<"YES\n"
#define NO cout<<"NO\n"
#define QYES quit("YES")
#define QNO quit("NO")
#define cmin(a, b) a = min(a, b)
#define cmax(a, b) a = max(a, b)
#define int long long
#define double long double


const int inf = numeric_limits<int>::max() / 4;
mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());
const int MOD = 1e9+7, LOG=20;

void solve();

signed main() {
   ios::sync_with_stdio(false);
   cin.tie(0);
   int t = 1;
   //cin >> t;
   while (t--) solve();
   return 0;
}

int k, n, m, p, q;
vector<vector<int>> adj;

void solve() {
    cin >> n >> m >> k;
    vector<string> a(n);
    cin.ignore();
    for (int i = 0; i < n; i++) getline(cin, a[i]);
    
    vector<vector<int>> cnt(n, vector<int>(n, 0));
    vector<vector<vector<int>>> letter(m, vector<vector<int>>(4));
    
    map<char,int> mp = {{'A',0},{'C',1},{'T',2},{'G',3}};

    vector<int> ok(n, 0);
    for (int i = 0; i < m; i++) {
        for (int j = 0; j < n; j++) letter[i][mp[a[j][i]]].pb(j);
    }

    for (int i = 0; i < m; i++) {
        int index = 0, sz = n+1;
        for (int j = 0; j < 4; j++) {
            if (sz(letter[i][j]) < sz && sz(letter[i][j])) {
                index = j;
                sz = sz(letter[i][j]);
            }
        }
        for (int x : letter[i][index]) {
            for (int j = 0; j < 4; j++) {
                if (j == index) continue;
                for (int y : letter[i][j]) {
                    cnt[y][x]++;
                    if (++cnt[x][y] == k) ok[x]++, ok[y]++;
                    else if (cnt[x][y] == k+1) ok[x]--, ok[y]--;
                }
            }
        }
    }

    int ans = 1;
    for (int i= 0; i < n; i++) if (ok[i]==n-1) ans += i;
    cout << ans << '\n';
}

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