UVALive - 7715 Guessing Game

点击打开题目

题意：给出A1,A2,B1,B2,C1,C2,N,问有多少种方法使得A1<=A<=A2,B1<=B<=B2,C1<=C<=C2且A+B+C=N

 

分析：要使A+B+C = N,即求A+B = N-C有多少种解法，那么N-C有一个取值范围[l, r],且这个区间的一个值唯一对应一个C值，那么现在就是求A+B = K(K = N-C)有多少种取值了

先看这样一个例子, A1 = 0, A2 = 2,B1 = 1, B2 = 6的情况，对于[A1, A2]每个值加上B的值的情况如下

A=0     A=1     A=2   对应的和     对应的和的AB的取法

 

              8        8          1

        7     7        7          2

 6      6     6        6          3

 5      5     5        5          3

 4      4     4        4          3

 3      3     3        3          3

 2      2              2          2

 1                     1          1

 可以发现，在[A1+B1,A2+B1)之间，对应的和的取法为等差数列，且公差为1，在[A2+B1, A1+B2]之间，对应的和的取法不变，在(A1+B2,A2+B2]之间，对应的和的取法为等差数列，公差为-1，暴力枚举各个区间就可以了...

 

注意: 如果数据量比较小, 用FFT求解会更简便.

#include<bits/stdc++.h>
typedef long long ll;
const int maxn = 3e3 + 10;
const ll mod = 1e9 + 7;
using namespace std;

ll qmod(ll a, ll b) {
    ll ans = 1;
    while(b) {
        if(b & 1) ans = ans * a % mod;
        a = a * a % mod;
        b >>= 1;
    }
    return ans;
}

int T, kase = 1;
ll A1, A2, B1, B2, C1, C2, N;

int main() {
    scanf("%d", &T);
    while(T--) {
        scanf("%lld %lld %lld %lld", &A1, &A2, &B1, &B2);
        scanf("%lld %lld %lld", &C1, &C2, &N);
        if(B2 - B1 < A2 - A1) { swap(A1, B1); swap(A2, B2); }
        ll L1 = A1 + B1, L2 = A2 + B1;
        ll L3 = B2 + A1, L4 = A2 + B2;
        ll L = N - C2, R = N - C1;
        ll ans = 0;
        ll a1 = ((L2 - L1) * (L2 - L1 + 1) % mod) * qmod(2, mod - 2) % mod;
        ll a2 = (L3 - L2 + 1) * (A2 - A1 + 1) % mod;
        ll a3 = a1;
        if(L > L4 || R < L1) ans = 0;
        else if(L < L1) {
            if(R >= L1 && R < L2) {
                ans = ((R - L1 + 1) * (R - L1 + 2) % mod) * qmod(2, mod - 2) % mod;
            } else if(R >= L2 && R <= L3) {
                ans = a1 % mod;
                ans = (ans + (R - L2 + 1) * (A2 - A1 + 1) % mod) % mod;
            } else if(R > L3 && R <= L4) {
                ans = (a1 + a2) % mod;
                ans = ((ans + a3 - ((L4 - R) * (L4 - R + 1)) % mod * qmod(2, mod - 2)) + mod) % mod;
            } else {
                ans = (a1 + a2 + a3) % mod;
            }
        } else if(L >= L1 && L < L2) {
            if(R < L2) {
                ll aa1 = (L - L1 + 1) % mod;
                ll n = (R - L + 1) % mod;
                ans = (n * aa1 + (n * (n - 1) % mod) * qmod(2, mod - 2)) % mod;
            } else if(R >= L2 && R <= L3) {
                ll aa1 = (L - L1 + 1) % mod;
                ll n = (L2 - L) % mod;
                ans = (n * aa1 + (n * (n - 1) % mod) * qmod(2, mod - 2)) % mod;
                ans = (ans + (R - L2 + 1) * (A2 - A1 + 1) % mod) % mod;
            } else if(R > L3 && R <= L4) {
                ll aa1 = (L - L1 + 1) % mod;
                ll n = (L2 - L) % mod;
                ans = (n * aa1 + (n * (n - 1) % mod) * qmod(2, mod - 2)) % mod;
                ans = (ans + a2) % mod;
                ans = ((ans + a3 - ((L4 - R) * (L4 - R + 1) % mod * qmod(2, mod - 2))% mod ) % mod + mod ) % mod;
            } else {
                ll aa1 = (L - L1 + 1) % mod;
                ll n = (L2 - L) % mod;
                ans = (n * aa1 % mod + (n * (n - 1) % mod) * qmod(2, mod - 2)) % mod;
                ans = (ans + a2) % mod;
                ans =(ans + a3) % mod;
            }
        } else if(L >= L2 && L <= L3) {
            if( R <= L3) {
                ans = (R - L + 1) * (A2 - A1 + 1) % mod;
            } else if(R > L3 && R <= L4) {
                ans = (L3 - L + 1) * (A2 - A1 + 1) % mod;
                ans = (ans + a3 - ((L4 - R) * (L4 - R + 1) % mod) * qmod(2, mod - 2) + mod) % mod;
            } else {
                ans = (L3 - L + 1) * (A2 - A1 + 1) % mod;
                ans = (ans + a3) % mod;
            }
        } else if(L > L3 && L <= L4){
            if(R <= L4) {
                ll aa1 = (L4 - R + 1) % mod;
                ll n = (R - L + 1) % mod;
                ans = (n * aa1 % mod + (n * (n - 1) % mod) * qmod(2, mod - 2)) % mod;
            } else {
                ans = ((L4 - L + 2) * (L4 - L + 1) % mod) * qmod(2, mod - 2) % mod;
            }
        }
        printf("Case #%d: %lld\n", kase++, ((ans % mod) + mod) % mod);
    }
    return 0;
}