bzoj3907: 网格

http://www.cnblogs.com/Tunix/p/4354348.html

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<iostream>

using namespace std;

void setIO(const string& s) {
    freopen((s + ".in").c_str(), "r", stdin);
    freopen((s + ".out").c_str(), "w", stdout);
}
template<typename Q> Q read(Q& x) {
    static char c, f;
    for(f = 0; c = getchar(), !isdigit(c); ) if(c == '-') f = 1;
    for(x = 0; isdigit(c); c = getchar()) x = x * 10 + c - '0';
    if(f) x = -x;
    return x;
}
template<typename Q> Q read() {
    static Q x; read(x); return x;
}

struct LL {
    const static int N = 5000, base = 10;
    int da[N], n;
    
    void init(int n) {
        this->n = n;
        memset(da, 0, sizeof(da[0]) * n);
    }
    
    void clear_zero() {
        while(n && !da[n-1]) n--;
    }
    
    LL operator * (const int &rhs) const {
        static LL c;
        c.init(n + 5);
        for(int i = 0; i < c.n; i++) {
            if(i < n) c.da[i] += da[i] * rhs;
            c.da[i+1] += c.da[i] / base;
            c.da[i] %= base;
        }
        c.clear_zero();
        return c;
    }
    
    void print() const {
        if(n == 0) putchar('0');
        else for(int i = n - 1; i >= 0; i--) {
            putchar(da[i] + '0');
        }
    }
}res;

const int LIM = 10000, N = 5000;

int mi[LIM+1], primes[N], tot;
bool flag[LIM+1];

void get_primes(int n) {
    for(int i = 2; i <= n; i++) {
        if(!flag[i]) primes[tot] = i, mi[i] = tot++;
        for(int j = 0; j < tot; j++) {
            int k = primes[j];
            if(i * k > n) break;
            flag[i * k] = 1;
            if(i % k == 0) break;
        }
    }
}

int num[N];

void add(int x, int sign) {
    for(int i = 0, k; k = primes[i], k * k <= x; i++) {
        while(x % k == 0) {
            num[i] += sign;
            x /= k;
        }
    }
    if(x ^ 1) num[mi[x]] += sign;
}

int main() {
#ifdef DEBUG
    freopen("in.txt", "r", stdin);
    freopen("out.txt", "w", stdout);
#endif
    
    int n, m;
    scanf("%d%d", &n, &m);
    get_primes(n + m);
    for(int i = n + m; i > m; i--) add(i, 1);
    for(int i = n; i > 1; i--) add(i, -1);
    add(n - m + 1, 1), add(n + 1, -1);
    
    res.da[0] = res.n = 1;
    
    for(int i = 0; i < tot; i++) {
        while(num[i]--) res = res * primes[i];
    }
    
    res.print(); puts("");
    
    return 0;
}