CodeForces - 103D(分块+dp)

题意:给你n个数,p个查询,每次查询下标为a,a+b,a+2b……的和。

思路:对n分块,b>\sqrt{n}时,可以直接暴力求和,最大扫描次数为O(\sqrt{n});b<=\sqrt{n}时,用一个dp[j]存放从第j个数出发求得的和,容易得出dp[j]=dp[j+b]+a[j](这里设a为数据数组,下面的a和题意相同),为了更优化,更新dp之前先判断这一次询问的b是不是和上一次相同,如果相同,那么可以直接使用上一次得出的dp,还要判断dp[a]是不是为0,不是0说明已经记忆过,直接输出即可。整体的复杂度就是O(n*\sqrt{n})。

因为我的疏忽把小于号重载写错了,一直runtime error on test 9!!

#include
using namespace std;
#define inf 0x3f3f3f3f
#define ll long long
const int maxn=300005;
const int mod=1e9+7;
const double eps=1e-8;
const double PI = acos(-1.0);
#define lowbit(x) (x&(-x))
struct query
{
    ll id,b,a;
    bool operator<(query x)
    {
        if(b!=x.b)  return bblock)
        {
            for(int j=q[i].a; j<=n; j+=q[i].b)
            {
                res+=a[j];
            }
        }
        else
        {
            if(q[i].b!=preb||dp[q[i].a]==0)
            {
                for(int j=n; j>=1; j--)
                {
                    if(j+q[i].b>n)  dp[j]=a[j];
                    else
                    {
                        dp[j]=dp[j+q[i].b]+a[j];
                    }
                }
            }
            res=dp[q[i].a];
        }
        ans[q[i].id]=res;
        preb=q[i].b;
    }
}
int main()
{
    std::ios::sync_with_stdio(false);
    cin>>n;
    for(int i=1; i<=n; i++)
    {
        cin>>a[i];
    }
    cin>>m;
    for(int i=0; i>q[i].a>>q[i].b;
        q[i].id=i;
    }
    sort(q,q+m);
    build();
    for(int i=0; i

 

你可能感兴趣的:(分块,DP)