Codeforces Round #512 C - Vasya and Golden Ticket

最新推荐文章于 2019-06-29 15:53:21 发布

原创最新推荐文章于 2019-06-29 15:53:21 发布 · 325 阅读

0 ·

CC 4.0 BY-SA版权

codeforces 专栏收录该内容

6 篇文章

订阅专栏

本文介绍了一种通过计算前缀和来判断字符串是否能被有效分割的算法。算法首先计算字符串的前缀和，然后尝试将字符串分割成多个部分，每个部分的前缀和必须能够整除整个字符串的总和。此外，还特别处理了字符串中包含0的情况。

在这里插入图片描述
Examples
input

5
73452

output
YES

input

4
1248

output

NO

水题= =
算一下前缀和，根据总和sum[n]，从2->n枚举一下可以分成几段。
假设分成k段，则遍历一下前缀和数组，若有k个sum[i]满足sum[i]%(sum[n]/k)==0，则改分割方法可行。
还有就是要特判一下字符0的情况。

#include<iostream>
#include<cstring>
#include<cstdio>
#include<string>
#include<vector>
#include<algorithm>
#include<cmath>

using namespace std;
int n;
char s[110];
int sum[110];

int solve(int num)
{
    int ans=0;
    int k=sum[n-1]/num;
    for(int i=0;i<n;i++)
    {
        if(s[i]!='0'&&sum[i]%k==0)
            ans++;
    }
    if(ans>=num) return 1;
    else return 0;
}
int main()
{
    int real_num=0;
   scanf("%d",&n);
   scanf("%s",s);
   int x=s[0]-'0';
   sum[0]=x;
   if(x!=0) real_num++;
   for(int i=1;i<n;i++)
   {
       x=s[i]-'0';
       if(x!=0) real_num++;
       sum[i]=sum[i-1]+x;
   }

   int flag=0;
   if(real_num==0) flag=1;
    if(!flag)
    {
        for(int i=2; i<=n; i++)
        {
            if(sum[n-1]%i==0)
            {
                int ans=solve(i);
                if(ans)
                {
                    flag=1;
                    break;
                }

            }
        }
    }

   if(flag) printf("YES\n");
   else printf("NO\n");
   return 0;
}