AcWing
  • 首页
  • 活动
  • 题库
  • 竞赛
  • 应用
  • 更多
    • 题解
    • 分享
    • 商店
    • 问答
    • 吐槽
  • App
  • 登录/注册

AcWing 3739. 农夫约翰合牛国

作者: 作者的头像   炽热的 ,  2023-01-13 21:32:35 ,  所有人可见 ,  阅读 58


2


枚举右端点 $i$, 然后左端点 $j$ 的有效范围在 $(last[b[i]] + 1, i)$ 才能保证该区间不存在和右端点 $i$ 相同的颜色,接着看左端点 $j$ 啥时候合法。很显然,左端点 $j$ 的右边没有和它一样的颜色,意味着这个 $j$ 是颜色 $b[j]$ 最后出现的位置,意味着一个合法的左端点 $j$。

于是确定右端点 $i$ 后,累加上 $(last[b[i]] + 1, i)$ 之间 $j$ 即可。

#include <cstring>
#include <iostream>
#include <algorithm>

using namespace std;

const int N = 2e5 + 10;

typedef long long i64;

int n;
int b[N], last[N];
int tr[N];

void add(int x, int v) {
    for (; x <= n; x += x & -x) tr[x] += v;
}

int query(int x) {
    int res = 0;
    for (; x; x -= x & -x) res += tr[x];
    return res;
}

int main() {
    scanf("%d", &n);
    for (int i = 1; i <= n; i ++ ) scanf("%d", &b[i]);

    i64 ans = 0;
    for (int i = 1; i <= n; i ++ ) {
        int v = b[i];
        if (last[v]) add(last[v], -1);
        ans += query(i) - query(last[v]);
        add(i, 1), last[v] = i;
    }

    printf("%lld\n", ans);

    return 0;
}

1 评论


用户头像
先wa一发签个到   9天前         踩      回复

真的强


你确定删除吗?

© 2018-2023 AcWing 版权所有  |  京ICP备17053197号-1
用户协议  |  隐私政策  |  常见问题  |  联系我们
AcWing
请输入登录信息
更多登录方式: 微信图标 qq图标 qq图标
请输入绑定的邮箱地址
请输入注册信息