UVa:1368 DNA Consensus String

贪心,只要保证和每一位上出现次数最多的字母相同即可。如果有多个次数最多相同的则选择字典序最小的。

 

#include <iostream>
#include <cstdlib>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <queue>
#include <vector>
#include <map>
#include <stack>
#include <algorithm>
#define MAXN 1005
#define MOD 1000000007
#define INF 1000000
#define ll long long
using namespace std;
struct Letter
{
    char name;
    int cnt;
    Letter(char c,int a):name(c),cnt(a) {}
    bool operator < (const Letter &p) const
    {
        if(cnt==p.cnt) return name<p.name;
        return cnt>p.cnt;
    }
};
int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        int n,L;
        scanf("%d%d",&n,&L);
        char str[1005];
        vector<Letter> vec[1005];
        for(int i=0; i<L; ++i)
        {
            vec[i].push_back(Letter('A',0));
            vec[i].push_back(Letter('C',0));
            vec[i].push_back(Letter('G',0));
            vec[i].push_back(Letter('T',0));
        }
        for(int i=0; i<n; ++i)
        {
            scanf("%s",str);
            for(int j=0; str[j]; ++j)
            {
                if(str[j]=='A') vec[j][0].cnt++;
                else if(str[j]=='C') vec[j][1].cnt++;
                else if(str[j]=='G') vec[j][2].cnt++;
                else if(str[j]=='T') vec[j][3].cnt++;
            }
        }
        vector<char> ans;
        int sum=0;
        for(int i=0; i<L; ++i)
        {
             sort(vec[i].begin(),vec[i].end());
             ans.push_back(vec[i][0].name);
             sum+=(n-vec[i][0].cnt);
        }
        for(int i=0;i<L;++i)
            printf("%c",ans[i]);
        printf("\n%d\n",sum);
    }
    return 0;
}


 

你可能感兴趣的:(贪心)