HDU 6020 MG loves apple

枚举每个非零点i为最高位，判断剩余的数中能不能删出符合 （0*a+1*b+2*c)%3==sum(剩余数之和)%3 && a+b+c==k-i 的情况

对于n==k+1的情况需要特判，只有一个0属于合法情况，虽然比赛的时候没过...拿这个hack了好多人

#include<bits/stdc++.h>
#define pb push_back
#define fr first
#define sc second
using namespace std;

typedef long long ll;
const int N = 1e5+5;

char s[N];
int f[3], t[3];

bool check(int k) {
    if(t[1]%3+t[2]%3 > k) return 0;
    if(t[1]+t[2] <= k) return t[0]+t[1]+t[2]>=k?1:0;
    int tt = t[1]+t[2]-(t[1]+t[2]-k)/3*3;
    if(tt > k) tt -= 3;
    return tt+t[0]>=k?1:0;
}

int main() {
    int T;
    cin >> T;
    while(T--) {
        memset(f, 0, sizeof(f));
        int n, k, bias = 0;
        bool flag = 0;
        scanf("%d%d%s", &n, &k, s);
        if(n-k == 1) {
            for(int i = 0; i < n; i++) {
                if((s[i]-'0')%3 == 0) { flag = 1; break; }
            }
        }

        for(int i = n-1; i>=0 && !flag; i--) {
            bias = (bias+s[i]-'0')%3;
            if(s[i] != '0') {
                if(i <= k) {
                    int m = k-i;
                    for(int b = 0; b < 3; b++)
                    for(int c = 0; c < 3; c++) if((b+2*c)%3 == bias){
                        if(f[1]<b || f[2]<c) continue;
                        t[1] = (f[1]-b)/3*3+b, t[2] = (f[2]-c)/3*3+c, t[0] = f[0];
                        if(check(m)) { flag = 1; break; }
                    }
                }
            }
            f[(s[i]-'0')%3]++;
        }
        printf("%s\n", flag?"yes":"no");
    }
    return 0;
}