ACM新手DAY 19 前缀和与差分

题解

A - Color the ball

又是涂气球，题目和昨天的一样

• 前缀和。
  思维和树状数组那天的那到题一样。只对区间的左边的点和右边的点的下一个点做修改。

#include<stdio.h>
#include<string.h>
int a[100005];
int main()
{
    int n,x,y;
    while(scanf("%d",&n)&&n!=0)
    {
        memset(a,0,sizeof(a));
        for(int i=1; i<=n; i++)
        {
            scanf("%d%d",&x,&y);
            a[x]++;
            a[y+1]--;
        }
        printf("%d",a[1]);
        for(int i=2; i<=n; i++)
        {
            a[i]+=a[i-1];
            printf(" %d",a[i]);
        }
        printf("\n");
    }
    return 0;
}

B - Molly’s Chemicals（样题）

题目：给出n个数，和一个数k，求有多少个区间的和等于k的x次方，x从0到无穷。

• 思路：遍历k的指数x，区间和用前缀和处理。

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int maxn = 1e5+10;
int a[maxn];

map<ll,int >mp;
int main()
{
    int n,k;
    cin>>n>>k;
    for(int i = 0; i<n; i++)
    {
        cin>>a[i];
    }
    ll p = 1 ;
    ll ans = 0;
    while(abs(p)<1e15)
    {
        mp.clear();
        mp[0]=1;
        ll sum = 0;
        for(int i = 0; i<n; i++)
        {
            sum+=a[i];
            ans+=mp[sum-p];//ans是记录有多少个区间满足条件,这里的sum-p，与下面的语句对应生效
            mp[sum]++;
        }
        p *= k;
        if(p==1)break;//如果p=1，一圈就过完了
    }
    cout<<ans<<endl;
    return 0;
}