1524. 和が奇数となる部分配列の数(C言語だとわかりやすい)

整数配列 arr が与えられます。合計が奇数となる部分配列の数を返してください。

答えが非常に大きくなる可能性があるため、10^9 + 7 を法として結果を返してください。

例 1:

入力: arr = [1,3,5]
出力: 4
説明: すべての部分配列は [[1],[1,3],[1,3,5],[3],[3,5], [5] です。 ]]。
すべての部分配列の合計は [1,4,9,3,8,5] です。
奇数の合計には [1,9,3,5] が含まれるため、答えは 4 になります。
例 2:

入力: arr = [2,4,6]
出力: 0
説明: すべての部分配列は [[2],[2,4],[2,4,6],[4],[4,6],[ 6 ]] 。
すべての部分配列の合計は [2,6,12,4,10,6] です。
すべての部分配列の合計は偶数であるため、答えは 0 になります。
例 3:

入力: arr = [1,2,3,4,5,6,7]
出力: 16
例 4:

入力: arr = [100,100,99,99]
出力: 4
例 5:

入力: arr = [7]
出力: 1

分析: この質問は動的プログラミング + 配列について調べます。

まず、奇数 + 奇数 = 偶数、および奇数 + 偶数 = 奇数であることを知る必要があります。ビット演算子を使用して (arr [i ] & 1 ) == 1 を判断できます。true の場合は奇数であり、それ以外の場合は false です。

int numOfSubarrays(int* arr, int n){
    int MOD=(int) 1e9+7;
    int odd=0,even=0,cnt=0;
        for(int i=0;i<n;i++){
              // odd 表示以当前元素 k 结尾和为奇数子数组数目
            // even 表示以当前元素 k 结尾和为偶数子数组数目
            if((arr[i]& 1) == 1){
                int t = odd;
                // 偶数 + 奇数k = 奇数              
                odd = even + 1; // k 为奇数,  {k, 偶数1+k, 偶数2 + k, ...}
                even = t;       // 奇数 + 奇数k = 偶数
            }else{
                even++;  // k 为偶数,  {k, 偶数1+k, 偶数2 + k, ...}
            }

            cnt = (cnt + odd) % MOD; 
    }
  return cnt;
}

おすすめ

転載: blog.csdn.net/zhi6fui/article/details/124261441