首页 > 其他 > 详细

cf776c

时间:2018-12-30 17:03:13      阅读:189      评论:0      收藏:0      [点我收藏+]

这题用尺取法是怼不出来的。。。

一开始看到区间和等于k的幂,并且有负数,首先想到将前缀和排序后用尺取法,但因为排序后的前缀和次序是乱的,只适用带绝对值的情况(poj2566),所以无法做。

看了题解后发现其实是个简单的dp

//sum[r]=sum[l]+kk;
//ans+=mp[sum[i]-kk];其实是个dp 
#include<bits/stdc++.h>
using namespace std;
#define ll long long 
#define maxn 100005

int n,a[maxn];
ll ans,kk,k,sum[maxn];

void solve(ll kk){
    map<ll,int>mp;
    for(int i=0;i<=n;i++){
        mp[sum[i]]++;
        ans+=mp[sum[i]-kk];
    }
}

int main(){
    scanf("%d%lld",&n,&k);
    sum[0]=ans=0;
    for(int i=1;i<=n;i++){
        scanf("%d",&a[i]);
        sum[i]=sum[i-1]+a[i];
    } 
    if(k==1){
        solve(1);
        printf("%lld\n",ans);
        return 0;
    }
    if(k==-1){
        solve(1);solve(-1);
        printf("%lld\n",ans);
        return 0;
    }
    
    kk=1;
    while(kk<=(ll)1000000000*n){
        solve(kk);
        kk*=k;        
    }
    printf("%lld\n",ans);
    return 0;
} 

 

cf776c

原文:https://www.cnblogs.com/zsben991126/p/10199708.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!