[백준 C++] 1969. DNA

garden.97·2021년 12월 1일
0

백준 C++

목록 보기
5/28
post-thumbnail

문제 링크

문제

DNA란 어떤 유전물질을 구성하는 분자이다. 이 DNA는 서로 다른 4가지의 뉴클레오티드로 이루어져 있다(Adenine, Thymine, Guanine, Cytosine). 우리는 어떤 DNA의 물질을 표현할 때, 이 DNA를 이루는 뉴클레오티드의 첫글자를 따서 표현한다. 만약에 Thymine-Adenine-Adenine-Cytosine-Thymine-Guanine-Cytosine-Cytosine-Guanine-Adenine-Thymine로 이루어진 DNA가 있다고 하면, “TAACTGCCGAT”로 표현할 수 있다. 그리고 Hamming Distance란 길이가 같은 두 DNA가 있을 때, 각 위치의 뉴클오티드 문자가 다른 것의 개수이다. 만약에 “AGCAT"와 ”GGAAT"는 첫 번째 글자와 세 번째 글자가 다르므로 Hamming Distance는 2이다.

우리가 할 일은 다음과 같다. N개의 길이 M인 DNA s1, s2, ..., sn가 주어져 있을 때 Hamming Distance의 합이 가장 작은 DNA s를 구하는 것이다. 즉, s와 s1의 Hamming Distance + s와 s2의 Hamming Distance + s와 s3의 Hamming Distance ... 의 합이 최소가 된다는 의미이다.


입력

첫 줄에 DNA의 수 N과 문자열의 길이 M이 주어진다. 그리고 둘째 줄부터 N+1번째 줄까지 N개의 DNA가 주어진다. N은 1,000보다 작거나 같은 자연수이고, M은 50보다 작거나 같은 자연수이다.


출력

첫째 줄에 Hamming Distance의 합이 가장 작은 DNA 를 출력하고, 둘째 줄에는 그 Hamming Distance의 합을 출력하시오. 그러한 DNA가 여러 개 있을 때에는 사전순으로 가장 앞서는 것을 출력한다.


예제 입력 / 출력

// 예제 입력 1
5 8
TATGATAC
TAAGCTAC
AAAGATCC
TGAGATAC
TAAGATGT
// 예제 출력 1
TAAGATAC
7
// 예제 입력 2
4 10
ACGTACGTAC
CCGTACGTAG
GCGTACGTAT
TCGTACGTAA
// 예제 출력 2
ACGTACGTAA
6
// 예제 입력 3
6 10
ATGTTACCAT
AAGTTACGAT
AACAAAGCAA
AAGTTACCTT
AAGTTACCAA
TACTTACCAA
// 예제 출력3
AAGTTACCAA
12

알고리즘
각 DNA의 같은 위치에서 제일 많이 나온 알파벳을 선택하는 것이 Hamming Distance의 합이 가장 작은 DNA를 구할 수 있는 방법이다.
예를 들어 각 DNA의 첫 위치의 알파벳이 각각 A, A, G, C, A 라면 A는 3, C는 1, G는 1이기 때문에 A가 선택되어야 하고 Hamming Distance의 합을 구하려면 총 5개에서 A의 개수인 3을 빼주어야 한다.


#include <iostream>
#include <vector>
#include <algorithm>

using namespace std;

int main(void) {

	int N, M, max_index, num = 0;
	vector<string> dna;
	string temp;

	cin >> N >> M;

	for (int i = 0; i < N; i++) {
		cin >> temp;
		dna.push_back(temp);
	}

	// A -> C -> G -> T
	for (int i = 0; i < M; i++) {

		vector<int> hd{ 0, 0, 0, 0 };

		for (int j = 0; j < N; j++) {
			if (dna[j][i] == 'A') hd[0]++;
			else if (dna[j][i] == 'C') hd[1]++;
			else if (dna[j][i] == 'G') hd[2]++;
			else if (dna[j][i] == 'T') hd[3]++;
		}

		max_index = max_element(hd.begin(), hd.end()) - hd.begin();

		if (max_index == 0) cout << 'A';
		else if (max_index == 1) cout << 'C';
		else if (max_index == 2) cout << 'G';
		else cout << 'T';

		num += N - hd[max_index];
		hd.clear();
	}

	cout << "\n" << num;
}

profile
who wants to become a backend developer💪👩‍💻

0개의 댓글