Codeforces Round #685 (Div. 2) C. String Equality 思维

传送门

题意：给你一个原串和模式串，问你能否通过两种操作把原串变成模式串。
操作方法：
1.交换任意相邻字符。
2.将k长度的相同字符子串全+1。

思路：

对于操作1，相当于我们可以任意排序原串。

结合操作2，我们可以尽可能的先将原串中相同的字符聚集起来，然后看看模式串中对应字符的数量，如果原串中更多一点，我们就把多出来的全部丢给下一位字符（+1）。

又因为要k个k个地+1，所以我们看多出来的那些，不够凑到k个的，先用变量bios记录下来，说明我们要用到后面的字符和当前这些剩下的字符一起+1。

如果原串字符更少一些，那么前面的bios就起作用了，就在此时和前面说的剩余字符一起+1，同时更新bios。到最后判断一下bios是否等于0且字符个数全部相等即可。

详见代码注释。

view code

#include<iostream>
#include<string>
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<map>
#include <queue>
#include<sstream>
#include <stack>
#include <set>
#include <bitset>
#include<vector>
#define FAST ios::sync_with_stdio(false)
#define abs(a) ((a)>=0?(a):-(a))
#define sz(x) ((int)(x).size())
#define all(x) (x).begin(),(x).end()
#define mem(a,b) memset(a,b,sizeof(a))
#define max(a,b) ((a)>(b)?(a):(b))
#define min(a,b) ((a)<(b)?(a):(b))
#define rep(i,a,n) for(int i=a;i<=n;++i)
#define per(i,n,a) for(int i=n;i>=a;--i)
#define endl '\n'
#define pb push_back
#define mp make_pair
#define fi first
#define se second
using namespace std;
typedef long long ll;
typedef pair<ll,ll> PII;
const int maxn = 1e6+200;
const int inf=0x3f3f3f3f;
const double eps = 1e-7;
const double pi=acos(-1.0);
const int mod = 1e9+7;
inline int lowbit(int x){return x&(-x);}
ll gcd(ll a,ll b){return b?gcd(b,a%b):a;}
void ex_gcd(ll a,ll b,ll &d,ll &x,ll &y){if(!b){d=a,x=1,y=0;}else{ex_gcd(b,a%b,d,y,x);y-=x*(a/b);}}//x=(x%(b/d)+(b/d))%(b/d);
inline ll qpow(ll a,ll b,ll MOD=mod){ll res=1;a%=MOD;while(b>0){if(b&1)res=res*a%MOD;a=a*a%MOD;b>>=1;}return res;}
inline ll inv(ll x,ll p){return qpow(x,p-2,p);}
inline ll Jos(ll n,ll k,ll s=1){ll res=0;rep(i,1,n+1) res=(res+k)%i;return (res+s)%n;}
inline ll read(){ ll f = 1; ll x = 0;char ch = getchar();while(ch>'9'||ch<'0') {if(ch=='-') f=-1; ch = getchar();}while(ch>='0'&&ch<='9') x = (x<<3) + (x<<1) + ch - '0',  ch = getchar();return x*f; }
int dir[4][2] = { {1,0}, {-1,0},{0,1},{0,-1} };

ll a[500];
ll b[500];
string s;
string t;

bool check(ll k)
{
    mem(a,0); mem(b,0);
    for(int i=0; i<s.size(); i++) a[s[i]]++;    //记录两串字符个数
    for(int i=0; i<t.size(); i++) b[t[i]]++;
    int bios = 0;   //偏移量
    for(int i='a'; i<'z'; i++)  //比较两串的每个字符，每一步都要变成相等（因为只能从小到大变化，所以当前变不到相等以后就变不到了）
    {
        if(a[i] > b[i])    // 当前字符原串更多的话
        {
            int d = a[i] - b[i];    // 看看差多少
            a[i] -= d;      //全部减去
            a[i+1] += d;    //丢给下一位
            bios += d - (d/k)*k ;   //看看有多少是凑不到k个的（剩下来的），说明要利用到后面的字符一起+1
        }
        else        // 如果原串少了
        {
            int d = b[i] - a[i];    //看看少了多少
            if(bios >= d) bios -= d;    //能用bios弥补回来就问题不大
            else return false;  //否则直接false
        }
    }
    if(bios||a['z'] != b['z']) return false;    //比较判断
    return true;
}

int main()
{
    int kase;
    cin>>kase;
    while(kase--)
    {
        ll n = read(), k = read();
        cin>>s;
        cin>>t;
        puts(check(k)?"YES":"NO");
    }
    return 0;
}