HDU 6020 MG loves apple

来源：互联网发布：游戏同步器软件编辑：程序博客网时间：2024/05/16 18:52

枚举每个非零点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 secondusing 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;}

0 0