Yukicoder No.434 占い

,

http://yukicoder.me/problems/no/434

No.432 占い(Easy)もまったく同じコードで通る。 入出力の形式を除いてNo.435 占い(Extra)のそれと同じコード。 解説はそちらのものを見て。

本番はぎりぎり通らず。 しかも本番の432への提出は嘘やミスにまみれておりひどかった。

implementation

#include <iostream>
#include <algorithm>
#include <vector>
#include <tuple>
#include <cassert>
#define repeat(i,n) for (int i = 0; (i) < (n); ++(i))
#define repeat_from(i,m,n) for (int i = (m); (i) < (n); ++(i))
#define whole(f,x,...) ([&](decltype((x)) y) { return (f)(begin(y), end(y), ## __VA_ARGS__); })(x)
using namespace std;
template <class T> void setmax(T & a, T const & b) { if (a < b) a = b; }
template <typename T, typename X> auto vectors(T a, X x) { return vector<T>(x, a); }
template <typename T, typename X, typename Y, typename... Zs> auto vectors(T a, X x, Y y, Zs... zs) { auto cont = vectors(a, y, zs...); return vector<decltype(cont)>(x, cont); }
pair<int,int> zero(int k) {
    int z = 0;
    while (k % 3 == 0) { k /= 3; z += 1; }
    return { k, z };
}
const int inv[9] = { 0, 1, 5, 0, 7, 2, 0, 4, 8 };
int choose(int n, int r) {
    static vector<pair<int,int> > fact(1, make_pair(1, 0));
    if (fact.size() <= n) {
        int l = fact.size();
        fact.resize( n + 1);
        repeat_from (i,l,n+1) {
            int pk, pz; tie(pk, pz) = fact[i-1];
            int k, z; tie(k, z) = zero(i);
            fact[i] = { pk * k % 9, pz + z };
        }
    }
    r = min(r, n - r);
    int pk, pz; tie(pk, pz) = fact[n];
    int qk, qz; tie(qk, qz) = fact[r];
    int rk, rz; tie(rk, rz) = fact[n-r];
    int p = pk * inv[qk] * inv[rk] % 9;
    int z = pz - qz - rz;
    assert (z >= 0);
    return z == 0 ? p : z == 1 ? p * 3 % 9 : 0;
}
int main() {
    int t; cin >> t;
    while (t --) {
        string s; cin >> s;
        int n = s.length();
        int acc = 0;
        repeat (j,n) acc = (acc + (s[j] - '0') * choose(n-1, j)) % 9;
        if (n != whole(count, s, '0') and acc == 0) acc = 9;
        cout << acc << endl;
    }
    return 0;
}