C. Did We Get Everything Covered?

给你两个整数 n 和 k 以及一个字符串 s 。
您的任务是检查是否所有长度为 n 的字符串都可以用前 k 个小写英文字母组成,并作为 s 的子序列出现。如果答案是否定的,那么您还需要打印一个长度为 n 的字符串,该字符串可以用前 k 个小写英文字母组成,但不会作为 s 的子序列出现。
如果有多个答案,您可以打印任意一个。
注: 如果从 b 中删除一些字符(可能为零)而不改变其余字符的顺序,就可以得到 a ,那么字符串 a 就被称为字符串 b 的子串。

输入:
第一行输入包含一个整数 t (1 ≤ t ≤ 1e5) ,即测试用例的数量。
每个测试用例的第一行包含 3 个整数 n(1 ≤ n ≤ 26),k(1 ≤ k ≤ 26),m(1 ≤ m ≤ 1000) ,其中 n 和 k 与输入中的描述相同, m 是字符串 s 的长度。
每个测试用例的第二行包含一个长度为 m 的字符串 s ,其中只有前 k 个小写英文字母。
保证所有测试用例的 m 和 n 之和不超过 1e6 。

输出:
对于每个测试用例,如果用前 k 个小写英文字母组成的长度为 n 的所有可能字符串都出现在 s 的子序列中,则打印 "YES",否则打印 "NO"。
如果答案为 "NO",请在下一行中打印一个长度为 n 的字符串,该字符串可以用第一个 k 小写英文字母组成,但不会作为 s 的子序列出现。

Input
3
2 2 4
abba
2 2 3
abb
3 3 10
aabbccabab

Output
YES
NO
aa
NO
ccc

注:
对于第一个测试用例,所有可能的长度为 2 的字符串(aa、ab、ba、bb),只要能用前 2 个英文字母组成,都会作为 abba 的子序列出现。长度为 2 的所有可能字符串(aa、ab、ba、bb)。
对于第二个测试用例,字符串 aa 不是 abb 的子序列。

解析:
容易想到在第一次出现的字母后,其它字母在后面必须出现至少 n-1 次;在第二次出现的字母后,其它字母在后面必须出现至少 n-2 次,以此类推。
因此就是在 s 中寻找 包含 k 个字母的循环节的出现次数,是否出现 n 次。
大于等于 n 次的,输出 “YES”;
不足 n 次的,输出 “NO”,每个循环节的最后一个字母+最后一次循环节未出现的字母 一定就是不属于 s 的子序列。

#include <bits/stdc++.h>
using namespace std;
#define int long long
#define endl '\n'
#define ios ios::sync_with_stdio(false),cin.tie(nullptr),cout.tie(nullptr);
int gcd(int a,int b) { return b? gcd(b,a%b) : a; }
typedef pair<int,int> PII;
const double PI=acos(-1.0);
const int N=2e6+10;
int n,k,m;
string s;
set <int> q;
int a[26];
vector <char> ans;
void solve()
{
    cin>>n>>k>>m;
    cin>>s;
    int cnt=0;
    q.clear(),ans.clear();
    memset(a,0,sizeof a);
    for (int i=0;i<m;i++)
    {
        q.insert(s[i]);
        if (q.size()==k)
        {
            cnt++;
            ans.push_back(s[i]);
            q.clear();
        }
    }
    if (cnt>=n) cout<<"YES\n";
    else 
    {
        cout<<"NO\n";
        for (auto x:q) a[x-'a']++;
        for (int i=0;i<26;i++)
        {
            if(a[i]==0)
            {
                for (auto x:ans) cout<<x;
                for (int j=ans.size();j<n;j++) cout<<char('a'+i);
                cout<<endl;
                break;
            }
        }
    }
}
signed main()
{
    ios;
    int T=1;
    cin>>T;
    while (T--) solve(); 
    return 0;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值