【牛客】寒假训练营4 E-漂亮数组 题解

传送门：漂亮数组
标签：思维

题目大意

阿宁认为一个数组是漂亮数组，该数组需要存在一个总和是k的倍数的子数组。现在阿宁有一个长度为n的数组a，阿宁想要将数组α分割出若干个数组。分割出的数组需要满足，按照分割顺序合并可以得到原数组a。阿宁想知道将数组α分割，最多可以获得多少个漂亮数组?
输入：第一行两个正整数n,k，第二行n个正整数a_i(1<=n<=2e5,1<=k,a_i<=1e9)，含义如题。
输出：一个整数，代表最多可以获得的漂亮数组数量。

算法分析

• 看完题我们就知道要找出所给数组中总和是k的倍数的子数组数量。但是由于同一元素可能出现在不同子数组中，所以符合条件的子数组并不是固定的，我们需要使其数量最大化。显然漂亮数组的数量就是这些子数组的数量，因为其余的元素怎么分配并不影响答案。
• 这时我们可能会想到用动态规划，但是看数据范围即使加上取模也有1e9，数组不可能开的下。所以说明想复杂了，仔细一看这题其实和最大子段和有点类似，但不同之处在于最大子段和能通过sum<0判断出当前情况不优从而舍弃前面的部分，而这道题似乎没有一个很清晰的标准来判断优劣。
• 不妨从贪心的角度考虑。如果下标从1到i-1的子数组不是漂亮数组，而下标从1到i是一个漂亮数组，那么我们可以把第i+1到n的元素当成一个新的数组去计算答案，再加上1。这个策略的正确性很容易证明，因为同一个元素虽然在不同的分配方式中可能出现中多个子数组中，但对于每一种确定的划分它都只可能被一个子数组包括。所以当前i个元素中只有一个符合条件的子数组且它以a_i结尾时，将这一段作为一个漂亮数组是最优策略。
• 那么问题就转换为如何判断子数组[a_i,a_i+1,a_i+2,…,a_j]是否为一个漂亮数组。我们先化简问题，假设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;
}