# | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
---|---|---|---|---|---|---|---|
438121 | josiftepe | DNA 돌연변이 (IOI21_dna) | C++17 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <iostream>
#include <vector>
#include <algorithm>
#include "dna.h"
using namespace std;
const int maxn = 1e5 + 5;
int a_pref_sum[maxn][4];
int b_pref_sum[maxn][4];
string a, b;
vector<int> a_id[4], b_id[4];
int n;
void init(string a, string b) {
memset(a_pref_sum, 0, sizeof a_pref_sum);
memset(b_pref_sum, 0, sizeof b_pref_sum);
n = (int) a.size();
for(int i = 0; i < n; i++) {
if(a[i] == 'C') a[i] = 'B';
if(b[i] == 'C') b[i] = 'B';
}
for(int i = 0; i < n; i++) {
if(a[i] == 'T') a[i] = 'C';
if(b[i] == 'T') b[i] = 'C';
}
::a = a;
::b = b;
int cnt_a[3] = {0, 0, 0};
int cnt_b[3] = {0, 0, 0};
for(int i = 0; i < n; i++) {
int id_a = (int) (a[i] - 'A');
int id_b = (int) (b[i] - 'A');
a_id[id_a].push_back(i);
b_id[id_b].push_back(i);
cnt_a[id_a]++;
cnt_b[id_b]++;
for(int j = 0; j < 3; j++) {
a_pref_sum[i][j] = cnt_a[j];
b_pref_sum[i][j] = cnt_b[j];
}
}
}
int a_query(int &x, int &y, char c) {
if(x == 0) {
return a_pref_sum[y][c - 'A'];
}
return a_pref_sum[y][c - 'A'] - a_pref_sum[x - 1][c - 'A'];
}
int b_query(int &x, int &y, char c) {
if(x == 0) {
return b_pref_sum[y][c - 'A'];
}
return b_pref_sum[y][c - 'A'] - b_pref_sum[x - 1][c - 'A'];
}
int get_distance(int x, int y) {
if(a_query(x, y, 'A') == b_query(x, y, 'A') and a_query(x, y, 'B') == b_query(x, y, 'B') and a_query(x, y, 'C') == b_query(x, y, 'C')) {
vector<int> v[3];
string tmp_b = b.substr(x, y - x + 1);
for(int i = x, j = 0; i <= y; i++, j++) {
v[b[i] - 'A'].push_back(j);
}
int cnt[3] = {0, 0, 0};
int ret = 0;
for(int i = x, j = 0; i <= y; i++, j++) {
int id_a = (a[i] - 'A');
int id_b = (tmp_b[j] - 'A');
if(id_a != id_b) {
swap(tmp_b[j], tmp_b[v[id_a][cnt[id_a]]]);
cnt[id_a]++;
++ret;
}
}
return ret;
}
return -1;
}