CodeForce 1215B. The Number of Products

最新推荐文章于 2024-06-04 22:50:27 发布

CharmsGraker

最新推荐文章于 2024-06-04 22:50:27 发布

阅读量58

点赞数

分类专栏：笔记文章标签：算法图论 c++

本文链接：https://blog.csdn.net/qq_35332805/article/details/129309699

版权

笔记专栏收录该内容

18 篇文章 0 订阅

订阅专栏

题目链接

CodeForce 1215B. The Number of Products

思路

因为数据规模有 $2·10^5$ ，所以类似 $d p [i] [j] [k]$ 的动态规划的 $O(n^2)$ 肯定不行。

但是又得统计区间，那么只能用哈希表了。这样一个区间的乘积就可被 $p re [i] / p re [j]$ 所表示。
由于只关心正负，那么把a[i]部用1或-1代表即可。
只不过需要在遍历前补足1个+1，不然第一个元素看不到自己能够形成的区间。+1不影响乘积的正负，故补+1
此外还需注意的是，答案先输出负的，再输出正的

#include<bits/stdc++.h>

using namespace std;
typedef long long LL;

int main() {
    int n;
    cin >> n;
    vector<int> a(n);
    unordered_map<int, LL> mp(2);
    vector<LL> ans(2);

    for (int i = 0; i < n; ++i) {
        LL v;
        cin >> v;
        a[i] = (v > 0 ? 1 : -1);
    }
    int pre = 1;
    mp[-1] = 0;
    mp[1] = 1;

    for (int i = 0; i < n; ++i) {
        pre = pre * a[i];
//        printf("mp[1]=%d, ", mp[1]);
//        printf("mp[-1]=%d\n", mp[-1]);
        if (pre > 0) {
            ans[0] += mp[1];
            ans[1] += mp[-1];
        } else {
            ans[0] += mp[-1];
            ans[1] += mp[1];
        }
        mp[pre]++;
//        printf("ans = %d %d\n", ans[0], ans[1]);
    }
    printf("%lld %lld", ans[1], ans[0]);
}