正确答案(字符串处理)

正确答案

题目描述:
小 H 与小 Y 刚刚参加完 UOIP 外卡组的初赛,就迫不及待的跑出考场对答案。
“吔,我的答案和你都不一样!”,小 Y 说道,”我们去找神犇们问答案吧”。
外卡组试卷中共有 m 道判断题,小 H 与小 Y 一共从其他 n 个神犇那问了答案。之后又从小 G 那里得知, 这 n 个神犇中有 p 个考了满分, q 个考了零分, 其他神犇不为满分或零分。
这可让小 Y 与小 H 犯了难。你能帮助他们还原出标准答案吗?如有多解则输出字典序最小的那个。无解输出-1。
输入格式:
第一行四个整数 n, m, p, q,意义如上描述。
接下来 n 行,每一行 m 个字符’N’或’Y’,表示这题这个神犇的答案。
输出格式:
仅一行,一个长度为 m 的字符串或是-1。
样例输入:
2 2 2 0
YY
YY
样例输出:
YY
数据范围:
30%:n<=100.
60%:n<=5000,m<= 100.
100%:1<=n<=30000,1<=m<=500.0<=p,q且p+q<=n.
注意:
scanf无法读入string,但是cin非常慢(貌似三倍的关系),所以scanf一个char[]然后根据长度,把char[]中的元素挨个赋值到string,这样做虽然看上去慢,但实测还是比cin快。
本题中还手动模拟了,根据字典序找字符串的全排列。

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<map>
using namespace std;
const int maxn=30010;
int n,m,r,w,tot,a[maxn],f[maxn];
char in[maxn];
bool ifgo;
string s[maxn];
map<string,int> g;
bool judge(int x,int y)
{
    int len1=s[x].size();
    int len2=s[y].size();
    for(int i=0;i<len1;i++)
    if((s[x][i]=='N'&&s[y][i]=='N')||
      (s[x][i]=='Y'&&s[y][i]=='Y'))
      return 0;
    return 1;
}
bool can(int x)
{
    string t=s[x];
    int sum=0;
    for(int i=0;i<t.size();i++)
    {
        if(t[i]=='N')
        t[i]='Y';
        else t[i]='N';
    }
    for(int i=1;i<=n;i++)
    if(t==s[i]) sum++;
    if(sum!=0)  return 0;
    return true;
}
void work1()
{
    string t1,t2;
    for(int i=0;i<m;i++)
    t1+='N',t2+='Y';
    while(1)
    {
        if(!g[t1]&&!g[t2])
        {
            cout<<t1;
            ifgo=1;
            return;
        }
        int flag=-1;
        for(int i=m-1;i>=0;i--)
        {
            if(t1[i]=='N')
            {
                flag=i;
                break;
            }
        }
        if(flag==-1)
        {
            printf("-1");
            ifgo=1;
            return;
        }
        t1[flag]='Y',t2[flag]='N';
        for(int i=flag+1;i<m;i++)
        t1[i]='N',t2[i]='Y';
    }
}
void work2()
{
    for(int i=1;i<=tot;i++)
    if(a[i]==r)
    {
        if(can(f[i]))
        {
            cout<<s[f[i]];
            ifgo=1;
            return;
        }
    }
}
void work3()
{
    for(int i=tot;i>=1;i--)
    if(a[i]==w)
    {
        if(can(f[i]))
        {
            for(int j=0;j<s[f[i]].size();j++)
            {
                if(s[f[i]][j]=='N')
                printf("Y");
                else printf("N");
            }
            ifgo=1;
            return;
        }
    }
}
void work4()
{
    for(int i=1;i<=tot;i++)
    if(a[i]==r)
    {
        for(int j=1;j<=tot;j++)
        {
            if(i==j) continue;
            if(a[j]==w)
            if(judge(f[i],f[j]))
            {
                cout<<s[f[i]];
                ifgo=1;
                return;
            }
        }
    }
    printf("-1");
    return;
}
int main()
{
    freopen("answer.in","r",stdin);
    freopen("answer.out","w",stdout);
    scanf("%d%d%d%d",&n,&m,&r,&w);
    for(int i=1;i<=n;i++)
    {
        scanf("%s",&in);
        for(int j=0;j<m;j++)
        s[i]+=in[j];
        g[s[i]]++;
    }
    sort(s+1,s+n+1);
    a[1]=1,tot=1,f[1]=1;
    for(int i=2;i<=n;i++)
    {
        if(s[i]==s[i-1])
        a[tot]++;
        else a[++tot]=1,f[tot]=i;
    }
    if(w==0&&r==0)       work1();
    else if(w==0&&!ifgo) work2();
    else if(r==0&&!ifgo) work3();
    else if(!ifgo)       work4();
    return 0;
}
原文地址:https://www.cnblogs.com/cax1165/p/6070863.html