Balanced Ternary String【Codeforces Round #531 (Div. 3)D】【贪心、构造】

题目链接


一道简单的构造,我们可以分成几个状态,因为所有的状态只有8个,所以,直接写每个状态即可,哎…… 被hack了…… 烦啊…… 谁让我写的好烂…… 好菜啊…… 呜呜呜


#include <iostream>
#include <cstdio>
#include <cmath>
#include <string>
#include <cstring>
#include <algorithm>
#include <limits>
#include <vector>
#include <stack>
#include <queue>
#include <set>
#include <map>
#define lowbit(x) ( x&(-x) )
#define pi 3.141592653589793
#define e 2.718281828459045
using namespace std;
typedef unsigned long long ull;
typedef long long ll;
const int maxN = 3e5 + 7;
int N, num0, num1, num2, splay, head0, head1, head2, sum0, sum1, sum2;
char a[maxN];
int N_0[maxN], N_1[maxN], N_2[maxN];
int main()
{
    scanf("%d", &N);
    head0 = head1 = head2 = 1;
    num0 = num1 = num2 = 0;
    scanf("%s", a + 1);
    for(int i=1; i<=N; i++)
    {
        if(a[i] == '0') N_0[++num0] = i;
        else if(a[i] == '1') N_1[++num1] = i;
        else N_2[++num2] = i;
    }
    sum0 = num0;    sum1 = num1;    sum2 = num2;
    splay = N/3;    //平衡点的中间值
    if(sum0 == splay && sum1 == splay && sum2 == splay) { printf("%s\n", a+1); return 0; }
    if(sum0 > splay && sum1 <= splay && sum2 <= splay)  //001
    {
        for(int i=splay+1; i<=sum0; i++)
        {
            if(sum1 < splay)
            {
                a[N_0[i]] = '1';
                sum1++;
            }
            else
            {
                a[N_0[i]] = '2';
                sum2++;
            }
        }
        printf("%s\n", a+1); return 0;
    }
    if(sum0 <= splay && sum1 > splay && sum2 <= splay)  //010
    {
        int det = sum1 - splay;
        while(det--)
        {
            if(sum0 < splay)
            {
                a[N_1[head1++]] = '0';
                sum0++;
            }
            else
            {
                a[N_1[num1--]] = '2';
                sum2++;
            }
        }
        printf("%s\n", a+1); return 0;
    }
    if(sum0 > splay && sum1 > splay && sum2 <= splay)   //011
    {
        for(int i=splay+1; i<=sum0; i++)
        {
            a[N_0[i]] = '2';
        }
        for(int i=splay+1; i<=sum1; i++)
        {
            a[N_1[i]] = '2';
        }
        printf("%s\n", a+1); return 0;
    }
    if(sum0 <= splay && sum1 <= splay && sum2 > splay)  //100
    {
        int det = sum2 - splay;
        while(det--)
        {
            if(sum0 < splay)
            {
                a[N_2[head2++]] = '0';
                sum0++;
            }
            else
            {
                a[N_2[head2++]] = '1';
                sum1++;
            }
        }
        printf("%s\n", a+1); return 0;
    }
    if(sum0 > splay && sum1 <= splay && sum2 > splay)   //101
    {
        for(int i=splay+1; i<=sum0; i++)
        {
            a[N_0[i]] = '1';
        }
        int det = sum2 - splay;
        while(det--)
        {
            a[N_2[head2++]] = '1';
            sum1++;
        }
        printf("%s\n", a+1); return 0;
    }
    if(sum0 <= splay && sum1 > splay && sum2 > splay)   //110
    {
        int det = sum1 - splay;
        while(det--)
        {
            a[N_1[head1++]] = '0';
            sum0++;
        }
        det = sum2 - splay;
        while(det--)
        {
            a[N_2[head2++]] = '0';
            sum0++;
        }
        printf("%s\n", a+1); return 0;
    }
    return 0;
}

 

  • 2
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

Wuliwuliii

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值