Codeforces Round #556 (Div. 2) C. Prefix Sum Primes

最新推荐文章于 2021-10-06 11:49:22 发布

weixin_34254823

最新推荐文章于 2021-10-06 11:49:22 发布

阅读量41

点赞数

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/csxaxx/p/10801378.html

博客介绍一道模拟题，题目要求改变数组排序使前缀和的素数最多。解题需判断1和2的个数，特判1个数为零、2个数为零、1只有一个且2个数不为0的情况，其余情况将1的个数分奇数和偶数考虑。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目大意让你改变数组的排序，使前缀和的素数最多；

这是一道模拟题，让你通过判断1和2的个数来解决，

只要特判一下1的个数是零的时候，2的个数是零的时候，或者1只有一个而2的个数又不是0个的时候，剩下的情况我们只有把1的个数分奇数和偶数来考虑，大致思路是这样，接下来就看代码吧

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
const int maxn = 200010;
int main()
{
    int a[maxn];
    int n;
    cin >> n;
    int ans1 = 0, ans2 = 0;
    for (int i = 0; i < n; i++)
    {
        scanf("%d", &a[i]);
        if (a[i] == 1)ans1++;
        else ans2++;
    }

     if (ans1 == 1 && ans2 !=0)
    {
        printf("2 1");
        for (int i = 1;i < ans2;i++)printf(" 2");
    }
    else if (ans1 == 0 && ans2 != 0)
    {
        for (int i = 0;i < ans2;i++)printf("2 ");
    }
    else if (ans2 == 0 && ans1 != 0)
    {
        for (int i = 0;i < ans1;i++)printf("1 ");
    }
    else
     {
         printf("2 1");
         if (ans1 % 2 == 1)
         {
             for (int i = 1;i < ans1;i++)printf(" 1");
             for (int i = 1;i < ans2;i++)printf(" 2");
         }
         else
         {
             for (int i = 1;i < ans1-1;i++)printf(" 1");
             for (int i = 1;i < ans2;i++)printf(" 2");
             printf(" 1");
         }
     }
}