1. 程式人生 > 實用技巧 >CF103D Time to Raid Cowavans(分塊)

CF103D Time to Raid Cowavans(分塊)

套路題,對於步長大於根號n的,直接暴力做,對於步長小於根號n的,保留離線後

對於每個步長for一遍,這樣複雜度也是n*根號n

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<int,int> pll;
const int N=5e5+10;
const int mod=1e9+7;
ll sum[N];
ll a[N];
vector<pll> num[N];
ll ans[N];
int main(){
    //ios::sync_with_stdio(false);
int n; cin>>n; int block=sqrt(n); int i; for(i=1;i<=n;i++){ scanf("%lld",&a[i]); } int m; cin>>m; for(i=1;i<=m;i++){ int x,y; scanf("%d%d",&x,&y); if(y>block){ for(int j=x;j<=n;j+=y){ ans[i]
+=a[j];//步長大的直接暴力 } } else{ num[y].push_back({i,x}); } } for(i=1;i<=block;i++){//不超過nsqrt(n) if(num[i].size()){ for(int j=n;j>=1;j--){ sum[j]=(j+i>n?0:sum[i+j])+a[j]; } for(auto x:num[i]){ ans[x.first]
=sum[x.second]; } } } for(i=1;i<=m;i++){ printf("%lld\n",ans[i]); } return 0; }
View Code