Submission #716240

#	Time	Username	Problem	Language	Result	Execution time	Memory
716240		kidlin	DNA 돌연변이 (IOI21_dna)	C++17	100 / 100	885 ms	10076 KiB

dna

#include <bits/stdc++.h>
 
using namespace std;
 
 
const int maxn = 1e5 + 10;
 
string s, t;
int pref[2][3][maxn];
int kf[3][3][maxn];
map < char, int > rev;
void init(std::string a, std::string b)
{
    s = a;
    t = b;
    rev['A'] = 0;
    rev['C'] = 1;
    rev['T'] = 2;
    for (int i = 0; i < a.size(); i ++)
    {
        if (i != 0)
        {
            for (int p = 0; p < 3; p ++)
                for (int k = 0; k < 2; k ++)
                    pref[k][p][i] = pref[k][p][i - 1];
 
            for (int p1 = 0; p1 < 3; p1 ++)
                for (int p2 = 0; p2 < 3; p2 ++)
                    kf[p1][p2][i] = kf[p1][p2][i - 1];
        }
        if (a[i] != b[i])
            kf[rev[a[i]]][rev[b[i]]][i] ++;
        pref[0][rev[a[i]]][i] ++;
        pref[1][rev[b[i]]][i] ++;
    }
}
 
int get_distance(int x, int y)
{
 
 
    string s1 = s.substr(x, y - x + 1);
    string t1 = t.substr(x, y - x + 1);
    int df[2][3], pf[3][3];
    for (int p = 0; p < 3; p ++)
    {
        df[0][p] = pref[0][p][y];
        df[1][p] = pref[1][p][y];
    }
 
    for (int p1 = 0; p1 < 3; p1 ++)
        for (int p2 = 0; p2 < 3; p2 ++)
            pf[p1][p2] = kf[p1][p2][y];
 
    if (x != 0)
    {
        for (int p = 0; p < 3; p ++)
        {
            df[0][p] -= pref[0][p][x - 1];
            df[1][p] -= pref[1][p][x - 1];
        }
 
        for (int p1 = 0; p1 < 3; p1 ++)
            for (int p2 = 0; p2 < 3; p2 ++)
                pf[p1][p2] -= kf[p1][p2][x - 1];
 
    }
    for (int p = 0; p < 3; p ++)
        if (df[0][p] != df[1][p])
            return -1;
    int ans = 0;
    for (int p1 = 0; p1 < 3; p1 ++)
        for (int p2 = 0; p2 < 3; p2 ++)
        {
            int mx = min(pf[p1][p2], pf[p2][p1]);
            ans += mx;
            pf[p1][p2] -= mx;
            pf[p2][p1] -= mx;
        }
 
    int add = 0;
    for (int p1 = 0; p1 < 3; p1 ++)
        for (int p2 = 0; p2 < 3; p2 ++)
            add += pf[p1][p2];
    ans = ans + (add / 3) * 2;
    return ans;
    }

Compilation message (stderr)

dna.cpp: In function 'void init(std::string, std::string)':
dna.cpp:19:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::__cxx11::basic_string<char>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   19 |     for (int i = 0; i < a.size(); i ++)
      |                     ~~^~~~~~~~~~

• Subtask 1: 21 / 21
• Subtask 2: 22 / 22
• Subtask 3: 13 / 13
• Subtask 4: 28 / 28
• Subtask 5: 16 / 16