Balanced Ternary String（贪心+思维）

题目链接：Balanced Ternary String

题目大意：给一个字符串，这个字符串只由0,1,2构成，然后让替换字符，使得在替换字符次数最少的前提下，使新获得的字符串中0,1,2

　　　　　　这三个字符的数目相同，并且新获得的字符串的字典序要尽可能的小；

直接数组做法：暴力遍历

 

/* */
# include <bits/stdc++.h>
using namespace std;
typedef long long ll;
 
int n;
int c[4];
string s;
int main()
{
    ios_base::sync_with_stdio(false);
    cin>>n>>s;
    memset(c, 0, sizeof(c));
 
    for(int i=0; i<n; i++ )
        c[s[i]-'0']++;
    for(int i=0; i<n&&c[0]<n/3; i++ )///0不够的时候从前往后换
    {
        if( c[s[i]-'0']<=n/3 )
            continue;
        if( s[i]!='0' )///只要不是0,且多就换
        {
            c[s[i]-'0']--;
            s[i] = '0';
            c[s[i]-'0']++;
        }
    }
 
    for(int i=n-1; i>=0&&c[2]<n/3; i-- )///2不够的时候从后往前换
    {
        if( c[s[i]-'0']<=n/3 )
            continue;
        if( s[i]!='2' )///只要不是2,且多就换
        {
            c[s[i]-'0']--;
            s[i] = '2';
            c[s[i]-'0']++;
        }
    }
 
    for(int i=0; i<n&&c[1]<n/3; i++ )///1不够的时候,从前往后换
    {
        if(c[s[i]-'0']<=n/3 )
            continue;
        if( s[i]=='2' )///先把多出来的2换了,不能换多的0,多的0要从后面换,否则字典序就大了
        {
            c[s[i]-'0']--;
            s[i]='1';
            c[s[i]-'0']++;
        }
    }
 
    for(int i=n-1; i>=0 && c[1]<n/3; i-- )///1还不够,从后往前换
    {
        if( c[s[i]-'0']<=n/3 )
            continue;
        if( s[i]!='1' )///只要不是1,并且多就换
        {
            c[s[i]-'0']--;
            s[i]='1';
            c[s[i]-'0']++;
        }
    }
    cout<<s<<endl;
    return 0;
}

 

用双向队列来做，跟数组做法一个思维，双向队列用来记录0,1,2的位置

/* */
# include <bits/stdc++.h>
using namespace std;
typedef long long ll;
deque<int>num[3];

int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    int n;
    cin>>n;
    string a;
    cin>>a;

    for(int i=0; i<n; i++ )
        num[a[i]-'0'].push_back(i);///向队列位加入0,1,2的位置,num[0]记录0的位置,num[1]记录1的位置,num[2]记录2的位置
    int k=n/3;
    if( num[0].size()==num[1].size() && num[1].size()==num[2].size())
        cout<<a<<endl;
    else
    {
        if( num[2].size()>k )///2多,从前往后,先换成0,再换成1
        {
            while( num[2].size()>k && num[0].size()<k )
            {
                int pos = num[2].front();
                num[2].pop_front();
                a[pos] = '0';
                num[0].push_front(pos);
            }
            while( num[2].size()>k && num[1].size()<k )
            {
                int pos = num[2].front();
                num[2].pop_front();
                a[pos] = '1';
                num[1].push_front(pos);
            }
        }

        if( num[1].size()>k )///1多,从前往后换成0,从后往前换成2
        {
            while( num[1].size()>k && num[2].size()<k )
            {
                int pos = num[1].back();
                num[1].pop_back();
                a[pos] = '2';
                num[2].push_back(pos);
            }

            while( num[1].size()>k && num[0].size()<k )
            {
                int pos=num[1].front();
                num[1].pop_front();
                a[pos] = '0';
                num[0].push_front(pos);
            }
        }

        if( num[0].size()>k )///0多,从后往前先换成2,再换成1
        {
            while( num[0].size()>k && num[2].size()<k )
            {
                int pos = num[0].back();
                num[0].pop_back();
                a[pos] = '2';
                num[2].push_back(pos);
            }

            while( num[0].size()>k && num[1].size()<k )
            {
                int pos = num[0].back();
                num[0].pop_back();
                a[pos] = '1';
                num[1].push_back(pos);
            }
        }
        cout<<a<<endl;
    }
    return 0;
}