题目链接:codeforces 792C. Divide by Three
今天队友翻了个大神的代码来问,我又想了遍这题,感觉很好,这代码除了有点长,思路还是清晰易懂,我就加点注释存一下。。。分类吧。删除一个数字模3为M的或删除两个模3为3-M的(还有一些要删零),具体看代码。
#include<cstdio> #include<cstring> #include<algorithm> #include<iostream> #include<string> #include<cmath> using namespace std; int L, M, i, j, W, A[100000]; bool C; string S, R; int main () { cin>>S; L=S.length(); for (i=0;i<L;i++) { A[i]=S[i]-'0'; M=(M+A[i])%3; } if (M==0) {//不用进行删除 cout<<S<<' '; return 0; } for (i=1;i<L;i++) {//从第二个数字开始 选择删除一个数字 if (A[i]%3==M) { for (j=0;j<i;j++) { cout<<A[j]; } for (j=i+1;j<L;j++) { cout<<A[j]; } cout<<' '; return 0; } } if (A[0]%3==M&&A[1]!=0) {//还是删除一个数字(模3为M),这种情况是第二个数字不为0并且可以删除第一个 for (j=1;j<L;j++) { cout<<A[j]; } cout<<' '; return 0; } for (i=L-1;i>=0;i--) {//删除两个数字(均是模3为3-M) 或更多 if (A[i]%3==3-M) { if (W) {//判断有两个可删数字 for (j=0;j<L;j++) { if (j!=W&&j!=i) {//删除两个及以上,可能有前导零 if (A[j]) { C=1; cout<<A[j]; } else if (C) { cout<<0; } } } if (!C) { if (L<3) { cout<<-1<<' '; } else { cout<<0<<' '; } } return 0; } else { W=i; } } } if (A[0]%3==M) {//删除一个数字或更多 这种情况是第一个数字是模3为M,第二个数字为0。这里先删除第一个数字 for (j=1;j<L;j++) { if (A[j]) {//可能有前导零 C=1; cout<<A[j]; } else if (C) { cout<<0; } } if (!C) { if (L==1) { cout<<-1<<' '; } else { cout<<0<<' '; } } } return 0; }
/*
//这代码的选择删除 模M 或 模3-M 的顺序很重要, 最后两种方法如果换个先后顺序就错了,比如这组数据
//20000111
//200001
//之前我的做法是同时进行两种删除方法,通过判断哪种方法删除数字更少,来选择输出更优的那个,而这个代码则不用比较两种情况,直接按这种顺序输出就行
*/