Codeforces 390C Inna and Candy Boxes RMQ简单变化 或 前缀和dp一下

题目大意：

就是对于给定的n, w, k (1 <= n, w <= 10^5, 1 <= k <= min(n, 10)), 和一串长度为n的只包含0和1的串, 接下来有w次询问每次输入L, R, 求在区间[L, R]中如果要求仅有位置L + k - 1, L + 2*k - 1, L + 3*k - 1, ... R 这些位置的数为1, 其他的为0, 需要变更多少个数

注意每次询问之后原来输入的串不改变

大致思路：

由于是静态的信息查询, 不需要使用线段树, 这里我用的是和RMQ类似的预处理的一个DP的方法做的...状态转移什么的都写在代码注释里了, 看到有很多人用的前缀和什么的做法..看来我还是想的复杂了一点....

代码如下：

Result  :  Accepted     Memory  :  8228 KB     Time  :  93 ms

/*
 * Author: Gatevin
 * Created Time:  2015/2/26 12:51:03
 * File Name: poi~.cpp
 */
#include<iostream>
#include<sstream>
#include<fstream>
#include<vector>
#include<list>
#include<deque>
#include<queue>
#include<stack>
#include<map>
#include<set>
#include<bitset>
#include<algorithm>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cctype>
#include<cmath>
#include<ctime>
#include<iomanip>
using namespace std;
const double eps(1e-8);
typedef long long lint;

/*
 * 用dp[i][j]表示从a[i]开始长度为(2^j)*k的连续段的数中需要改变的数的个数
 * 也就是区间[i, i + (2^j)*k - 1]中在询问以i开头时需要变更的数的数量
 * 和RMQ的预处理一样, 先预处理算出dp[1~n][0]
 * 然后递推dp[i][j] = dp[i][j - 1] + dp[i + (2^(j - 1))*k - 1][j - 1]
 * 每次询问区间[L, R]时根据长度的二进制来拆分成多个2^p (0 <= p)的区间的dp值得和即可
 */

/*
 * 当然看到别人用前缀和...表示我一看到这个题就想到和RMQ类似的dp方法于是直接这么写了...
 */
int n, w, k;
int dp[100010][20];
int a[100010];

void initRMQ()//预处理
{
    for(int i = 1; i <= n; i++)
    {
        if(i + k - 1 > n) continue;
        for(int j = i; j <= i + k - 1; j++)
        {
            if(j != i + k - 1 && a[j] == 1)
                dp[i][0]++;
            else if(j == i + k - 1 && a[j] == 0)
                dp[i][0]++;
        }
    }
    for(int j = 1; (1 << j)*k <= n; j++)
        for(int i = 1; i + (1 << j)*k - 1 <= n; i++)
            dp[i][j] = dp[i][j - 1] + dp[i + (1 << (j - 1))*k][j - 1];
    return;
}

int askRMQ(int l, int r)//询问区间[l, r]
{
    int len = r - l + 1;
    len /= k;
    if(len == 1) return dp[l][0];
    int ret = 0;
    int cnt = 0;
    while(len)//拆开区间为多个2的倍数的长度
    {
        if(len & 1)
        {
            ret += dp[l][cnt];
            l += (1 << cnt)*k;
        }
        cnt++;
        len >>= 1;
    }
    return ret;
}

int main()
{
    cin>>n>>k>>w;
    for(int i = 1; i <= n; i++)
        scanf("%1d", &a[i]);
    initRMQ();
    int l, r;
    for(int i = 1; i <= w; i++)
    {
        scanf("%d %d", &l, &r);
        printf("%d\n", askRMQ(l, r));
    }
    return 0;
}