Codeforces B. The Number of Products(遞推)

傳送門
思路:用遞推就好了(我沒有想到)。

#include<iostream>
using namespace std;

const int MAXN = 2e5+5;
long long dp1[MAXN], dp2[MAXN]; //以i爲結尾的,有dp1[i]個區間含有奇數個負數,dp2[i]個區間含有偶數個負數
long long a[MAXN];

int main(){
    int n;
    cin >> n;
    for(int i = 1; i <= n; ++i)
        cin >> a[i];
    dp1[1] = (int) (a[1] < 0), dp2[1] = 1-dp1[1];
    long long ans1 = dp1[1], ans2 = dp2[1];
    for(int i = 2; i <= n; ++i){
        if(a[i] < 0){
            dp1[i] = dp2[i-1]+1;
            dp2[i] = dp1[i-1];
        }
        else{
            dp1[i] = dp1[i-1];
            dp2[i] = dp2[i-1]+1;
        }
        ans1 += dp1[i];
        ans2 += dp2[i];
    }
    cout << ans1 << " " << ans2 << endl;
    return 0;
}
 
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章