*hiho 1475 - 数组拆分,dp,由N^2降到NlogN

题目链接

描述

小Ho得到了一个数组作为他的新年礼物,他非常喜欢这个数组!

在仔细研究了几天之后,小Ho成功的将这个数组拆成了若干段,并且每段的和都不为0!

现在小Ho希望知道,这样的拆分方法一共有多少种?

两种拆分方法被视作不同,当且仅当数组断开的所有位置组成的集合不同。

输入

每组输入的第一行为一个正整数N,表示这个数组的长度

第二行为N个整数A1~AN,描述小Ho收到的这个数组

对于40%的数据,满足1<=N<=10

对于100%的数据,满足1<=N<=105, |Ai|<=100

输出

对于每组输入,输出一行Ans,表示拆分方案的数量除以(109+7)的余数。

-----------------------------------------------------------------------------------------------------------

容易想到一个N^2的解法:

F[i] = Sum(F[0<=j<i])+(pre[i]!=0),pre[j]!=pre[i]. 其中pre[i]为前缀和。

因为N为10w,所以肯定过不了。

我们可以先不判断pre[j]是否等于pre[i],即维护一个累加和,然后再减去前缀和等于pre[i]的累加和:F[i] = 当前累加和 - pre[i]的累加和。

pre[i]的累加和用map维护,是logN的复杂度。

这样就由N^2降到了NlogN

#include <map>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
const int MOD = 1e9+7;
const int N = 100100;
int sum[N];
long long dp[N];
int main(){
    long long pre = 0;
    map<long,long> records;
    int n; cin>>n; for(int i=0;i<n;i++){
        scanf("%d",sum+i); if(i) sum[i]+=sum[i-1];
        dp[i] = sum[i]?((pre+1)%MOD):pre;
        auto iter = records.find(sum[i]);
        if(iter==records.end()) records[sum[i]]=dp[i];
        else {
            dp[i] = (dp[i]-iter->second+MOD)%MOD;
            iter->second = (iter->second+dp[i])%MOD;
        }
        pre = (pre+dp[i])%MOD;
    }
    printf("%lld\n",dp[n-1]);
    return 0;
}
时间: 2024-09-30 10:50:20

*hiho 1475 - 数组拆分,dp,由N^2降到NlogN的相关文章

数组拆分

温习并学习下算法,记录设计地点滴. 数组拆分是将一个无序数组,拆分成两个子数组,子数组A地元素全部比数组元素小,子数组B地元素全部比数组元素大. 代码如下: package test; import java.util.Arrays; public class PartitionPolicy { public void part(int[] array, int key) { // swap int temp = array[key]; array[key] = array[array.leng

java数组拆分多个子数组

这里做个java集合/数组拆分多个子集合/数组的两种实现方式的笔记. 方式 一: /**  * 拆分集合  * @param datas  * @param splitSize  * @param <T>  * @return  */ private <T> List<List<T>> spliceArrays(List<T> datas, int splitSize) {     if (datas == null || splitSize &

将数组拆分为字符串

//将数组拆分为字符串 $attr=$db->Query($sql); $str=""; foreach($attr as $v) { $str =$str. implode("^",$v); $str = $str."|"; } $str = substr($str,0,strlen($str)-1); echo $str; //将字符串拼接为数组 var hang = data.split("|"); for(var

HDU 6240 Server(2017 CCPC哈尔滨站 K题,01分数规划 + 树状数组优化DP)

题目链接  2017 CCPC Harbin Problem K 题意  给定若干物品,每个物品可以覆盖一个区间.现在要覆盖区间$[1, t]$. 求选出来的物品的$\frac{∑a_{i}}{∑b_{i}}$的最小值. 首先二分答案,那么每个物品的权值就变成了$x * b_{i} - a_{i}$ 在判断的时候先把那些权值为正的物品全部选出来, 然后记录一下从$1$开始可以覆盖到的最右端点的位置. 接下来开始DP,按照区间的端点升序排序(左端点第一关键字,右端点第二关键字) 问题转化为能否用剩

Leetcode#561. Array Partition I(数组拆分 I)

题目描述 给定长度为 2n 的数组, 你的任务是将这些数分成 n 对, 例如 (a1, b1), (a2, b2), ..., (an, bn) ,使得从1 到 n 的 min(ai, bi) 总和最大. 示例 1: 输入: [1,4,3,2] 输出: 4 解释: n 等于 2, 最大总和为 4 = min(1, 2) + min(3, 4). 提示: n 是正整数,范围在 [1, 10000]. 数组中的元素范围在 [-10000, 10000]. 思路 分组之后min(ai, bi)的和最大

HDU 6447 - YJJ&#39;s Salesman - [树状数组优化DP][2018CCPC网络选拔赛第10题]

Problem DescriptionYJJ is a salesman who has traveled through western country. YJJ is always on journey. Either is he at the destination, or on the way to destination.One day, he is going to travel from city A to southeastern city B. Let us assume th

LUOGU P2344 奶牛抗议 (树状数组优化dp)

传送门 解题思路 树状数组优化dp,f[i]表示前i个奶牛的分组的个数,那么很容易得出$f[i]=\sum\limits_{1\leq j\leq i}f[j-1]*(sum[i]\ge sum[j-1])$,但是这样的时间复杂度是$O(n^2)?$,所以考虑优化,发现必须满足$sum[i]\ge sum[j-1]?$才能进行转移,那么直接离散化后用树状数组维护一个前缀和即可. #include<iostream> #include<cstdio> #include<cstr

LeetCode561 数组拆分 I

给定长度为 2n 的数组, 你的任务是将这些数分成 n 对, 例如 (a1, b1), (a2, b2), ..., (an, bn) ,使得从1 到 n 的 min(ai, bi) 总和最大. 示例 1: 输入: [1,4,3,2] 输出: 4 解释: n 等于 2, 最大总和为 4 = min(1, 2) + min(3, 4). 提示: n 是正整数,范围在 [1, 10000]. 数组中的元素范围在 [-10000, 10000]. //章节 - 数组和字符串 //四.双指针技巧 //2

Leetcode 561.数组拆分I

数组拆分 I 给定长度为 2n 的数组, 你的任务是将这些数分成 n 对, 例如 (a1, b1), (a2, b2), ..., (an, bn) ,使得从1 到 n 的 min(ai, bi) 总和最大. 示例 1: 输入: [1,4,3,2] 输出: 4 解释: n 等于 2, 最大总和为 4 = min(1, 2) + min(3, 4). 提示: n 是正整数,范围在 [1, 10000]. 数组中的元素范围在 [-10000, 10000]. 思路 这道题目给了我们一个数组有2n i