Codeforces 746D Green And Black Tea 構造__CODEFORCES_

點擊開啟連結

題意:有n杯茶，a杯綠茶，b杯紅茶，問怎麼擺放才可以讓不超過k杯一樣的茶連續擺放，如果不能就輸出NO。

//顏色少的Y至少要有n/(k+1)個 插入後如果還有剩下,則在每組裡面隔一個間隔插入即可 

#include <bits/stdc++.h>using namespace std;typedef long long ll;const int N=2e5+20;char s[N];int main(){int n,k,a,b;cin>>n>>k>>a>>b;char X='G',Y='B';//預設較大的為X字元 if(a<b){swap(a,b);swap(X,Y);}for(int i=1;i<=n;i++)s[i]=X;for(int i=k+1;i<=n;i+=k+1){//每k+1個位置至少要一個不同字元 if(b==0){puts("NO");return 0;}s[i]=Y; b--;}//插入 b個Y 剩下的自然就是a個X for(int i=1;i<=n&&b;i++){ //每組中隔一個間隔插入 if(s[i]==Y||(i-1>=1&&s[i-1]==Y)||(i+1<=n&&s[i+1]==Y))continue;s[i]=Y;b--;}printf("%s",s+1);return 0;}