传送门:漂亮数组
标签:思维
题目大意
阿宁认为一个数组是漂亮数组,该数组需要存在一个总和是k的倍数的子数组。现在阿宁有一个长度为n的数组a,阿宁想要将数组α分割出若干个数组。分割出的数组需要满足,按照分割顺序合并可以得到原数组a。阿宁想知道将数组α分割,最多可以获得多少个漂亮数组?
输入:第一行两个正整数n,k,第二行n个正整数ai(1<=n<=2e5,1<=k,ai<=1e9),含义如题。
输出:一个整数,代表最多可以获得的漂亮数组数量。
算法分析
- 看完题我们就知道要找出所给数组中总和是k的倍数的子数组数量。但是由于同一元素可能出现在不同子数组中,所以符合条件的子数组并不是固定的,我们需要使其数量最大化。显然漂亮数组的数量就是这些子数组的数量,因为其余的元素怎么分配并不影响答案。
- 这时我们可能会想到用动态规划,但是看数据范围即使加上取模也有1e9,数组不可能开的下。所以说明想复杂了,仔细一看这题其实和最大子段和有点类似,但不同之处在于最大子段和能通过sum<0判断出当前情况不优从而舍弃前面的部分,而这道题似乎没有一个很清晰的标准来判断优劣。
- 不妨从贪心的角度考虑。如果下标从1到i-1的子数组不是漂亮数组,而下标从1到i是一个漂亮数组,那么我们可以把第i+1到n的元素当成一个新的数组去计算答案,再加上1。这个策略的正确性很容易证明,因为同一个元素虽然在不同的分配方式中可能出现中多个子数组中,但对于每一种确定的划分它都只可能被一个子数组包括。所以当前i个元素中只有一个符合条件的子数组且它以ai结尾时,将这一段作为一个漂亮数组是最优策略。
- 那么问题就转换为如何判断子数组[ai,ai+1,ai+2,…,aj]是否为一个漂亮数组。我们先化简问题,假设i=1。这样一来就可以开一个map<long long,bool>,将其命名为mp。mp[x]=1代表在前面存在一个前缀和对k取模后等于x。如果当前的前缀和取模后也为x就说明存在一段区间加起来为k的倍数。这时对答案加一,并舍弃前面所有元素,将后面一段看作一个全新的数组。具体操作为:清空map,并令前缀和sum=0。该算法总体时间复杂度为遍历的复杂度乘以map底层红黑树的复杂度,也就是O(nlogn)。
代码实现
#include <iostream>
using namespace std;
#include <map>
map<long long,bool> mp;
char g[1001][1001];
#define lowbit(x) (x)&(-x)
int main(){
long long k,p,q,i,j,x,y,x1,y1,n,m,T,sum,ans;
ios::sync_with_stdio(false);
cin.tie(0);cout.tie(0);
T=1;
while(T--){
cin>>n>>k;
sum=ans=0LL;
mp[0]=1;
for(i=1;i<=n;i++){
cin>>x;
sum=(sum+x)%k;
if(mp[sum]){ans++;mp.clear();sum=0LL;mp[0]=1;}
else mp[sum]=1;
}
cout<<ans;
}
return 0;
}