【HDU 6020】 MG loves apple （乱搞？）

MG loves apple

 Accepts: 20

 Submissions: 693

 Time Limit: 3000/1500 MS (Java/Others)

 Memory Limit: 262144/262144 K (Java/Others)

问题描述

MGMG是一个财富爆表的男孩子。他拥有N(1<=N<=100000)N(1<=N<=100000)个苹果，每个苹果上标有一个数字00~99，代表它的价值。

一个合法的数字是不含有前导零的，这nn个苹果恰好排成了一个合法的NN位数。

MGMG拥有拿去KK个苹果的权利（0<=K<0<=K<NN）。
他想知道是否存在方案，使得恰好拿去KK个苹果后，序列中剩下的苹果排成的合法数字模33等于零。数据保证所有NN之和不超过10000001000000.

MGMG认为这件事非常容易，不屑于用计算机解决，于是运用他高超的人类智慧开始进行计算。作为一名旁观者，你也想挑战MGMG智慧，请你写个程序，计算答案。

输入描述

第一行一个整数TT，代表数据组数（1 <=T<=601<=T<=60）。
接下来，对于每组数据——
第一行两个个整数NN,KK，表示苹果序列长度，以及需要拿去的苹果个数.
接下来一行NN个整数XX，表示每个苹果的价值（0<=X<=90<=X<=9）。

输出描述

对于每一组数据，输出一行。
若方案存在，则输出“yes”，否则输出“no”。（输出不包含引号）

输入样例

2
5 2
11230
4 2
1000 

输出样例

yes
no

【分析】

　　写这个主要是我昨天WA了7次没A。。

　　怎么说，打法太容易出bug。。

　　【表示不知道别人是怎么做的。。。

　　为了保证没有前导0，我们枚举一个不为0的开头，于是问题变成了询问在a0个0，a1个1，a2个2中选x个数，他们加上y能不能模3等于0。

　　这个东西啊，我是不敢枚举什么的。我们可以先随便选x个数，然后判断，不行的话，交换1个或者两个选了和没选的数，再判断。

　　显然交换的数的个数不会超过2个的哦。

　　这里我就暴力枚举交换的东西了。

　　【主要是我昨天没有枚举最前面的数，而是贪心先删掉后k个再交换，然后前面交换时的前导0没有了保证，就搞了很久。。。

　　【还是ORZ那些秒A的人

 #include<cstdio>
 #include<cstdlib>
 #include<cstring>
 #include<iostream>
 #include<algorithm>
 using namespace std;
 #define Maxn 100010

 int mymax(int x,int y) {return x>y?x:y;}
 int mymin(int x,int y) {return x<y?x:y;}

 int f[Maxn][];
 int b[],c[];

 bool check(int a0,int a1,int a2,int x,int y)
 {
     // if(x==0) return y==0;
     // if(x==1) return (y==0&&a0)||(y==1&&a1)||(y==2&&a2);
     int h=;
     b[]=b[]=b[]=c[]=c[]=c[]=;
     b[]=mymin(a0,x);c[]=a0-b[],x-=b[];
     b[]=mymin(a1,x);c[]=a1-b[],x-=b[],h=(h+b[])%;
     b[]=mymin(a2,x);c[]=a2-b[],x-=b[],h=(h+b[]*)%;
     if((h+y)%==) return ;

     for(int i=;i<;i++) if(b[i])
      for(int j=;j<;j++) if(c[j])
          if(((h-i+j+y)%+)%==) return ;

     for(int i=;i<;i++) if(b[i])
     {
         b[i]--;
         for(int ii=;ii<;ii++) if(b[ii])
         {
             for(int j=;j<;j++) if(c[j])
             {
                 c[j]--;
                 for(int jj=;jj<;jj++) if(c[jj])
                 {
                     if(((h-i-ii+j+jj+y)%+)%==) return ;
                 }
                 c[j]++;
             }
         }
         b[i]++;
     }
     return ;
 }

 int a[Maxn];
 char s[Maxn];

 int main()
 {
     int T;
     scanf("%d",&T);
     while(T--)
     {
         int n,k;
         scanf("%d%d",&n,&k);
         scanf("%s",s+);
         for(int i=;i<=n;i++) a[i]=s[i]-'';
         if(k==n-)
         {
             bool ok=;
             for(int i=;i<=n;i++) if(a[i]%==) ok=;
             if(ok) printf("yes\n");
             else printf("no\n");
             continue;
         }
         memset(f,,sizeof(f));
         for(int i=;i<=n;i++)
          for(int j=;j<;j++) f[i][j]=f[i-][j]+(a[i]%==j);
         k=n-k;
         bool ok=;
         for(int i=;i<=n-k+;i++) if(a[i]!=)
         {
             if(check(f[n][]-f[i][],f[n][]-f[i][],f[n][]-f[i][],k-,a[i]%)) {ok=;break;}
         }
         if(ok) printf("yes\n");
         else printf("no\n");
     }
     return ;
 }

2017-04-02 08:51:33