码迷,mamicode.com
首页 > 其他好文 > 详细

cf776c

时间:2018-12-30 17:29:24      阅读:215      评论:0      收藏:0      [点我收藏+]

标签:while   namespace   efi   amp   fine   排序   lld   scanf   ace   

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

一开始看到区间和等于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

标签:while   namespace   efi   amp   fine   排序   lld   scanf   ace   

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

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!