【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个的哦。

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

【HDU 6020】 MG loves apple (乱搞?)【HDU 6020】 MG loves apple (乱搞?)

 1 #include<cstdio>
 2 #include<cstdlib>
 3 #include<cstring>
 4 #include<iostream>
 5 #include<algorithm>
 6 using namespace std;
 7 #define Maxn 100010
 8 
 9 int mymax(int x,int y) {return x>y?x:y;}
10 int mymin(int x,int y) {return x<y?x:y;}
11 
12 int f[Maxn][3];
13 int b[3],c[3];
14 
15 bool check(int a0,int a1,int a2,int x,int y)
16 {
17     // if(x==0) return y==0;
18     // if(x==1) return (y==0&&a0)||(y==1&&a1)||(y==2&&a2);
19     int h=0;
20     b[0]=b[1]=b[2]=c[0]=c[1]=c[2]=0;
21     b[0]=mymin(a0,x);c[0]=a0-b[0],x-=b[0];
22     b[1]=mymin(a1,x);c[1]=a1-b[1],x-=b[1],h=(h+b[1])%3;
23     b[2]=mymin(a2,x);c[2]=a2-b[2],x-=b[2],h=(h+b[2]*2)%3;
24     if((h+y)%3==0) return 1;
25     
26     for(int i=0;i<3;i++) if(b[i])
27      for(int j=0;j<3;j++) if(c[j])
28          if(((h-i+j+y)%3+3)%3==0) return 1;
29      
30     for(int i=0;i<3;i++) if(b[i])
31     {
32         b[i]--;
33         for(int ii=0;ii<3;ii++) if(b[ii])
34         {
35             for(int j=0;j<3;j++) if(c[j])
36             {
37                 c[j]--;
38                 for(int jj=0;jj<3;jj++) if(c[jj])
39                 {
40                     if(((h-i-ii+j+jj+y)%3+3)%3==0) return 1;
41                 }
42                 c[j]++;
43             }
44         }
45         b[i]++;
46     }
47     return 0;
48 }
49 
50 int a[Maxn];
51 char s[Maxn];
52 
53 int main()
54 {
55     int T;
56     scanf("%d",&T);
57     while(T--)
58     {
59         int n,k;
60         scanf("%d%d",&n,&k);
61         scanf("%s",s+1);
62         for(int i=1;i<=n;i++) a[i]=s[i]-0;
63         if(k==n-1)
64         {
65             bool ok=0;
66             for(int i=1;i<=n;i++) if(a[i]%3==0) ok=1;
67             if(ok) printf("yes/n");
68             else printf("no/n");
69             continue;
70         }
71         memset(f,0,sizeof(f));
72         for(int i=1;i<=n;i++)
73          for(int j=0;j<3;j++) f[i][j]=f[i-1][j]+(a[i]%3==j);
74         k=n-k;
75         bool ok=0;
76         for(int i=1;i<=n-k+1;i++) if(a[i]!=0)
77         {
78             if(check(f[n][0]-f[i][0],f[n][1]-f[i][1],f[n][2]-f[i][2],k-1,a[i]%3)) {ok=1;break;}
79         }
80         if(ok) printf("yes/n");
81         else printf("no/n");
82     }
83     return 0;
84 }