Submission #848305

#	Time	Username	Problem	Language	Result	Execution time	Memory
848305		vjudge1	Genetics (BOI18_genetics)	C++17	100 / 100	295 ms	36788 KiB

genetics

#include <bits/stdc++.h>
#define pb push_back
#define fi first
#define se second
#define FOR(i,a,b) for(int i = (a) ; i <= (b) ; i++)
#define FORE(i,a,b) for(int i = (a) ; i >= (b) ; i--)
#define par pair<int , int>
#define int long long

using namespace std;

const int N = 1e5+10;
const int INF = INT_MAX;
const int mod = 1e9+7;
const int base = 101;

int n,m,k;
string st[N];
int lab[N],s[N][30],sum;
int pos[4];

main()
{
    if (fopen("solve.inp", "r")) {
        freopen("solve.inp", "r", stdin);
        freopen("solve.out", "w", stdout);
    }
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);
    cin >> n >> m >> k;
    lab[0] = 1;
    for(int i = 1 ; i <= n ; i++) {
        lab[i] = lab[i-1]*base%mod;
    }
    for(int i = 1 ; i <= n ; i++) {
        sum += lab[i];
        sum %= mod;
    }
    pos[0] = 'A' - 'A';
    pos[1] = 'C' - 'A';
    pos[2] = 'G' - 'A';
    pos[3] = 'T' - 'A';
    for(int i = 1 ; i <= n ; i++) {
        string w; cin >> w;
        w = " " + w;
        st[i] = w;
        for(int j = 1 ; j <= m ; j++) {
            s[j][w[j]-'A'] += lab[i];
            s[j][w[j]-'A'] %= mod;
        }
    }
    for(int i = 1 ; i <= n ; i++) {
        int ans = (sum-lab[i]+mod*mod)%mod*k%mod;
        int res = 0;
        for(int j = 1 ; j <= m ; j++) {
            int c = st[i][j]-'A';
            for(int p = 0 ; p < 4 ; p++) {
                int h = pos[p];
                if(h == c) {
                    continue;
                }
                res += s[j][h];
                res %= mod;
            }
        }
        if(res == ans) {
            cout << i;
            return 0;
        }
    }
    return 0;
}

Compilation message (stderr)

genetics.cpp:22:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
   22 | main()
      | ^~~~
genetics.cpp: In function 'int main()':
genetics.cpp:25:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   25 |         freopen("solve.inp", "r", stdin);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~
genetics.cpp:26:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   26 |         freopen("solve.out", "w", stdout);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~

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