cf C. Inna and Candy Boxes

题意：给你一个长度为n的只含有1和0的字符串，w个询问，每次询问输入l,r；在[l,r]中在l+k-1、l+2*k-1、......r的位置都必须为1，如果不为1的，变成1，记为一次操作，其它的地方的都必须为0，不为0的地方要变成1，也记为一次操作，最后问在区间[l,r]最少几次操作。

思路：可以用树状数组记录在某个地方的右方有多少个1，然后在 预处理出从1，2，..k的为开头的在i+c*k-1的位置上前面有多少个1，最后的答案是拿走的1+添加的1，拿走的1的个数等于现有的1的个数-位置正确的1的个数。 添加的1的个数等于需要添加的个数的总数-位置正确的1的个数。

 1 #include <cstdio>
 2 #include <cstring>
 3 #include <algorithm>
 4 #define maxn 100010
 5 using namespace std;
 6 
 7 int n,k,w;
 8 char str[maxn];
 9 int a[maxn];
10 int c[maxn];
11 int dp[maxn];
12 int sum[maxn];
13 
14 int lowbit(int x)
15 {
16     return x&-x;
17 }
18 
19 void insert(int x,int d)
20 {
21     while(x<maxn)
22     {
23        c[x]+=d;
24        x+=lowbit(x);
25     }
26 }
27 
28 int Getsum(int x)
29 {
30     int ans=0;
31     while(x>0)
32     {
33         ans+=c[x];
34         x-=lowbit(x);
35     }
36     return ans;
37 }
38 
39 
40 int main()
41 {
42     while(scanf("%d%d%d",&n,&k,&w)!=EOF)
43     {
44          scanf("%s",str+1);
45          memset(sum,0,sizeof(sum));
46          memset(a,0,sizeof(a));
47          memset(c,0,sizeof(c));
48          for(int i=1; i<=n; i++)
49          {
50             if(str[i]=='1')
51             {
52                 insert(i,1);
53             }
54          }
55          for(int i=1; i<=k; i++)
56          {
57              int cnt=0;
58              for(int c=1; i+c*k-1<=n; c++)
59              {
60                   cnt+=(str[i+c*k-1]-'0');
61                   dp[i+c*k-1]=cnt;
62              }
63          }
64          for(int i=1; i<=w; i++)
65          {
66              int l,r;
67              int ans=0;
68              scanf("%d%d",&l,&r);
69              int num=(r-l+1)/k;
70              ans+=Getsum(r)-Getsum(l-1);
71              ans-=dp[r]-dp[l-1];
72              ans+=num;
73              ans-=dp[r]-dp[l-1];
74              printf("%d\n",ans);
75          }
76     }
77     return 0;
78 }

View Code