Submission #691730

#	Time	Username	Problem	Language	Result	Execution time	Memory
691730		ismayil	Mutating DNA (IOI21_dna)	C++17	100 / 100	45 ms	7540 KiB

dna

#include <bits/stdc++.h>
#define ll long long
using namespace std;
const int MAX = 1e5 + 5;
int dp[MAX][3][3], cnt[MAX][3];
int n;
void init(string a, string b){
    map<char, char> m = {{'A', 0}, {'T', 1}, {'C', 2}};
    n = a.size();
    for (int i = 0; i < n; i++)
    {
        a[i] = m[a[i]];
        b[i] = m[b[i]];
    }
    
    dp[0][a[0]][b[0]]++;
    cnt[0][a[0]]++;
    cnt[0][b[0]]--;
    for (int i = 1; i < n; i++)
    {
        for (int j = 0; j < 3; j++){
            for (int k = 0; k < 3; k++) dp[i][j][k] = dp[i - 1][j][k];

            cnt[i][j] = cnt[i - 1][j];
        }
        dp[i][a[i]][b[i]]++;
        cnt[i][a[i]]++;
        cnt[i][b[i]]--;
    }
    
}
int get_distance(int x, int y){
    if(cnt[y][0] - (x ? cnt[x - 1][0] : 0) != 0) return -1;
    if(cnt[y][1] - (x ? cnt[x - 1][1] : 0) != 0) return -1;
    if(cnt[y][2] - (x ? cnt[x - 1][2] : 0) != 0) return -1;

    int c[3][3];
    for (int i = 0; i < 3; i++)
    {
        for (int j = 0; j < 3; j++)
        {
            c[i][j] = dp[y][i][j];
            if(x != 0) c[i][j] -= dp[x - 1][i][j];
        }
    }

    return (min(c[0][1], c[1][0]) + min(c[1][2], c[2][1]) + min(c[0][2], c[2][0]) + 2 * abs(c[0][1] - c[1][0]));
}

Compilation message (stderr)

dna.cpp: In function 'void init(std::string, std::string)':
dna.cpp:16:15: warning: array subscript has type 'char' [-Wchar-subscripts]
   16 |     dp[0][a[0]][b[0]]++;
      |               ^
dna.cpp:16:21: warning: array subscript has type 'char' [-Wchar-subscripts]
   16 |     dp[0][a[0]][b[0]]++;
      |                     ^
dna.cpp:17:16: warning: array subscript has type 'char' [-Wchar-subscripts]
   17 |     cnt[0][a[0]]++;
      |                ^
dna.cpp:18:16: warning: array subscript has type 'char' [-Wchar-subscripts]
   18 |     cnt[0][b[0]]--;
      |                ^
dna.cpp:26:19: warning: array subscript has type 'char' [-Wchar-subscripts]
   26 |         dp[i][a[i]][b[i]]++;
      |                   ^
dna.cpp:26:25: warning: array subscript has type 'char' [-Wchar-subscripts]
   26 |         dp[i][a[i]][b[i]]++;
      |                         ^
dna.cpp:27:20: warning: array subscript has type 'char' [-Wchar-subscripts]
   27 |         cnt[i][a[i]]++;
      |                    ^
dna.cpp:28:20: warning: array subscript has type 'char' [-Wchar-subscripts]
   28 |         cnt[i][b[i]]--;
      |                    ^

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