[2018 Multi-University Training Contest 4] HDU 6333 Problem B. Harvest of Apples(莫队算法)

最新推荐文章于 2018-08-29 10:49:41 发布

ZZZZone

最新推荐文章于 2018-08-29 10:49:41 发布

阅读量233

点赞数

本文链接：https://blog.csdn.net/ZZZZone/article/details/81433870

版权

HDU 同时被 3 个专栏收录

8 篇文章 0 订阅

订阅专栏

2018 Multi-University Training

2 篇文章 0 订阅

订阅专栏

莫队算法

1 篇文章 0 订阅

订阅专栏

该博客介绍了如何利用莫队算法解决HDU 6333题目中关于苹果收获的问题。通过将数据范围和解题思路进行分析，提出将m和n视为区间，应用莫队算法将时间复杂度降低到O(n*sqrt(n))。博主分享了首次使用莫队算法的心得，并提供了AC代码。

摘要由CSDN通过智能技术生成

题目链接

Problem B. Harvest of Apples

题目大意

$T$ 组询问, 每组询问问 $\sum_{i = 0} ^ {m} C_n^i$ .

数据范围

$1 \le T \le 10^5$
$1 \le m \le n \le 10^5$

解题思路

定义 $S(n,m) = \sum_{i = 0} ^ {m} C_n^i$ .
由杨辉三角的递推公式 $C_n^m = C_{n-1}^{m-1} + C_{n-1}^{m}$ 可以推导出 $S(n,m) = 2\times S(n-1, m) - C_{n-1}^m$

这样我们就可以得到下面四个转移公式:

S (n, m) = 2 \times S (n - 1, m) - C m n - 1 S (n, m) = S ( n + 1 , m ) + C ( n , m ) 2 S (n, m) = S (n, m - 1) + C (n, m) S (n, m) = S (n, m + 1) - C (n, m + 1)

$S(n,m) = 2\times S(n-1, m) - C_{n-1}^m\\[2ex] S(n,m) = \frac{S(n+1, m) + C(n, m)}{2}\\[2ex] S(n,m) = S(n, m - 1) + C(n, m) \\[2ex] S(n,m) = S(n, m + 1) - C(n, m + 1)$

即使这样可以 $O(1)$ 的转移，但是最坏也是需要 $O(n^2)$ 的复杂度，没办法实现。
这时候就可以把 $m$ 和 $n$ 看成区间 $l$ 和 $r$ 。使用莫队算法处理一下。以前没写过莫队，正好趁这次机会学会了莫队算法。

使用莫队，将 $n$ 分成 $\sqrt{n}$ 块，按照 $l$ 所在的块从小到大排序，对于同一块内，按照 $r$ 从小到大排序。
这样的话可以使复杂度降到 $O(n * \sqrt{n})$ 。
简单证明（现学现卖）：

对于同一块内， $l$ 最多移动 $\sqrt n$ , $r$ 最多移动 $n$ 次，总共有 $\sqrt n$ 块，所以这种情况的复杂度为 $O(n * \sqrt{n})$
对不在同一块内的， $l$ 最多移动 $2 \times \sqrt n$ 次， $r$ 还是移动 $n$ 次，所以还是 $O(n * \sqrt{n})$

AC代码

/********************************************
 *Author*        :ZZZZone
 *Created Time*  : 日  8/ 5 14:48:44 2018
 * Ended  Time*  : 日  8/ 5 15:24:14 2018
 * 莫队算法
*********************************************/

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <vector>
#include <queue>
#include <set>
#include <map>
#include <string>
#include <cmath>
#include <cstdlib>
#include <ctime>
#include <stack>
using namespace std;
#define debug(x) std::cerr << #x << " = " << (x) << std::endl
typedef pair<int, int> PII;
typedef long long LL;
typedef unsigned long long ULL;

inline void OPEN(string s){
    freopen((s + ".in").c_str(), "r", stdin);
    freopen((s + ".out").c_str(), "w", stdout);
}

const int MAXN = 1e5;
const int Mod = 1e9 + 7;
LL fac[MAXN+5], rfac[MAXN+5], Ans[MAXN+5];
int blk[MAXN+5];
int n, Block;

struct Query{
    int l, r, id;
    bool operator < (const Query& tmp) const{
        if(blk[l] == blk[tmp.l]) return r < tmp.r;
        return blk[l] < blk[tmp.l];
    }
}Q[MAXN+5];

LL fast_pow(LL x, LL y){
    LL res = 1LL;
    while(y != 0LL){
        if(y & 1) res = (res * x) % Mod;
        y >>= 1;
        x = x * x % Mod;
    }
    return res;
}

inline LL Comb(LL N, LL M){
    if(N<M)return 0LL; //  ******  一定不能忘  ******   //
    return fac[N] * rfac[M] % Mod * rfac[N-M] % Mod;
}

void Init(){
    fac[0] = 1LL;
    for(int i = 1; i <= MAXN; i++) fac[i] = fac[i-1] * (LL)i % Mod;
    for(int i = 0; i <= MAXN; i++){
        rfac[i] = fast_pow(fac[i], Mod - 2LL);
        blk[i] = i / Block;
    }
    for(int i = 1; i <= n; i++) scanf("%d %d", &Q[i].r, &Q[i].l), Q[i].id = i;
    sort(Q + 1, Q + 1 + n);
}

int main()
{
    scanf("%d", &n);
    Block = sqrt(n);
    Init();
    int L = 1, R = 0;
    LL ans = 1LL;
    for(int i = 1; i <= n; i++){
        while(L < Q[i].l){
            L++;
            ans = (ans + Comb(R, L)) % Mod;
        }
        while(L > Q[i].l){
            ans = (ans - Comb(R, L) + Mod) % Mod;
            L--;
        }
        while(R < Q[i].r){
            ans = (2LL * ans % Mod - Comb(R, L) + Mod) % Mod;
            R++;
        }
        while(R > Q[i].r){
            R--;
            ans = (ans + Comb(R, L)) % Mod * rfac[2] % Mod;
        }
        Ans[Q[i].id] = ans % Mod;
    }
    for(int i = 1; i <= n; i++) printf("%lld\n", Ans[i]);
    return 0;
}