Codeforces Round #685 (Div. 2) C. String Equality 思維
阿新 • • 發佈:2020-11-22
題意:給你一個原串和模式串,問你能否通過兩種操作把原串變成模式串。 操作方法: 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; }