算法竞赛入门经典 习题3-7

UVa1368

DNA Consensus String

本来以为题目是要在已有序列中找一个最小的序列，后来仔细读了下题才发现是要构造一个距离最小的序列。

感觉这道题应该就是贪心算法吧？每一位和已有序列的对应位差得越少，总体也就越少。

#include <iostream>
#include <vector>
#include <string>
#include <array>

using namespace std;

int main()
{
	int T = 0;
	cin >> T;
	char ch[4] = { 'A', 'C', 'G', 'T' };
	for (int t = 0; t < T; t++)
	{
		int m = 0, n = 0, distance = 0;
		cin >> m >> n;
		vector<string> vstrDNA(m, "");
		for (int i = 0; i < m; i++)
		{
			cin >> vstrDNA[i];
		}
		string strDNA;
		array<int, 4> arrCnt;
		for (int i = 0; i < n; i++)
		{
			arrCnt = { { 0, 0, 0, 0 } };
			for (int j = 0; j < m; j++)
			{
				if (vstrDNA[j][i] == 'A') arrCnt[0]++;
				else if (vstrDNA[j][i] == 'C') arrCnt[1]++;
				else if (vstrDNA[j][i] == 'G') arrCnt[2]++;
				else if (vstrDNA[j][i] == 'T') arrCnt[3]++;
			}
			int idx = 0;
			for (int k = 0; k < 4; k++)
			{
				if (arrCnt[k] > arrCnt[idx]) idx = k;
			}
			strDNA.push_back(ch[idx]);
			distance += m - arrCnt[idx];
		}
		cout << strDNA << endl << distance << endl;
	}
	return 0;
}
/*
3
5 8
TATGATAC
TAAGCTAC
AAAGATCC
TGAGATAC
TAAGATGT
4 10
ACGTACGTAC
CCGTACGTAG
GCGTACGTAT
TCGTACGTAA
6 10
ATGTTACCAT
AAGTTACGAT
AACAAAGCAA
AAGTTACCTT
AAGTTACCAA
TACTTACCAA
*/