Codeforces 103D Time to Raid Cowavans

来源：互联网发布：mac 装虚拟机编辑：程序博客网时间：2024/06/01 23:45

题意：

给你n<=3*1e5头重w[i]的牛和p<=3*1e5个询问，每个询问给出a、b，问你所有a、a+b、a+2*b........a+k*b<=n的牛的重量。

思路：

由于这里的n、p都很大，我们直接暴力的话会TLE，那么我们先将b从小到大排序，分块处理（为什么要分呢？其实我也不知道）。分成mid=sqrt(n+1)，若b>=mid，直接暴力，若b<mid，则定义dp[j]为以j为起点now_b为等差的累加和值，即对于一个以a为起点b为等差的数来说，如果我们已经知道了dp[a]的值那么我们可以直接得到答案，否则从后往前推，dp[j]=dp[j+b]+w。

#include<cstdio>#include<cmath>#include<cstring>#include<algorithm>using namespace std;typedef __int64 LL;const int MAX=300005;int n,m;LL w[MAX],ans[MAX],dp[MAX];struct Node{int id,a,b;}c[MAX];bool cmp(const Node& t1,const Node& t2){if(t1.b!=t2.b) return t1.b<t2.b;else return t1.a<t2.a;}int main(){scanf("%d",&n);for(int i=1;i<=n;i++){scanf("%I64d",&w[i]);}scanf("%d",&m);for(int i=0;i<m;i++){scanf("%d%d",&c[i].a,&c[i].b);c[i].id=i;}sort(c,c+m,cmp);int now_b=0;int mid=sqrt(n+1);memset(ans,0,sizeof(ans));memset(dp,0,sizeof(dp));for(int i=0;i<m;i++){if(c[i].b>=mid){for(int j=c[i].a;j<=n;j+=c[i].b){ans[c[i].id]+=w[j];}}else{if(c[i].b!=now_b||dp[c[i].a]==0){for(int j=n;j>=1;j--){if(j+c[i].b>n) dp[j]=w[j];else dp[j]=dp[j+c[i].b]+w[j];}}ans[c[i].id]=dp[c[i].a];now_b=c[i].b;}}for(int i=0;i<m;i++){printf("%I64d\n",ans[i]);}return 0;}

0 0