codeforces 792C. Divide by Three

Posted 2020-09-08 GraceSkyer

题目链接：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<<‘\n‘;
        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<<‘\n‘;
            return 0;
        }
    }
    if (A[0]%3==M&&A[1]!=0) {//还是删除一个数字(模3为M)，这种情况是第二个数字不为0并且可以删除第一个
        for (j=1;j<L;j++) {
            cout<<A[j];
        }
        cout<<‘\n‘;
        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<<‘\n‘;
                    }
                    else {
                        cout<<0<<‘\n‘;
                    }
                }
                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<<‘\n‘;
            }
            else {
                cout<<0<<‘\n‘;
            }
        }
    }
    return 0;
}

/*
//这代码的选择删除 模M 或 模3-M 的顺序很重要， 最后两种方法如果换个先后顺序就错了，比如这组数据
//20000111
//200001
//之前我的做法是同时进行两种删除方法，通过判断哪种方法删除数字更少，来选择输出更优的那个，而这个代码则不用比较两种情况，直接按这种顺序输出就行
*/