力扣2025.分割数组的最多方案数

力扣2025.分割数组的最多方案数

• 哈希表 + 前缀和

  • 用两个哈希表分别存元素(之后会遍历)左侧和右侧的前缀和
  • 

  typedef long long LL;
  class Solution {
  public:
      int waysToPartition(vector<int>& nums, int k) {
          int n = nums.size(),ans = 0;
          vector<LL> sum(n);
          unordered_map<LL,int> cnt,mp;
          sum[0] = nums[0];
          for(int i=1;i<n;i++)
          {
              sum[i] = sum[i-1] + nums[i];
              //全存到右侧
              cnt[sum[i-1]] ++;
          }
          LL tot = sum[n-1];
          //不改的解
          if(tot % 2 == 0) ans = cnt[tot/2];
          for(int i=0;i<n;i++)
          {
              //每个元素都修改试试
              int d = k - nums[i];
              //改完发现满足
              if((tot + d) % 2 == 0)
                  //左侧的满足的前缀和 + 右侧的满足的前缀和(左边的合法分割点+右边的)
                  ans = max(ans,mp[(tot + d) / 2] + cnt[(tot - d) / 2]);
              //右边 ++，左边--
              mp[sum[i]] ++;
              cnt[sum[i]] --;
          }
          return ans;
      }
  };