E - Divisible by 3 Gym - 103102E

29 篇文章 0 订阅

题目连接
题意:sum(bi*bj) (l<=i<j<=r)问从1~n有多少个这样的(l,r)满足以上公式计算出来的数%3==0。
思路:dp,我们发现这个题对于每一个i,我们可以记录一个前缀和要求的权值,因为权值很大,所以我们要来%3的权值,这样就可以状态压缩一下。
简单分析一下为什么要用到前缀和和权值,首先是权值,我们要权值是因为权值是我们最终要的答案是权值%3为0的,要前缀和是因为我们要用前缀和来控制他的L,因为我们记录的前缀和是从1~i的,然而这里并不一定是从1到i的还可能是从2 ~ i,所以我们要用到前缀和。这里我们发现,在%3之后,前缀和有3种状态,权值也有3种状态。所以,这里就直接记录不同状态的数量就行了,下面是是AC代码:

#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
using namespace std;
#define int long long
const int N = 5e5 + 10;
int dp[N][5][5];
int a[N];
signed main()
{
  int n;
  cin >> n;
  for (int i = 1; i <= n; i++)
  {
    cin >> a[i];
    dp[i][a[i]%3][0]++;
  }
  for (int i = 1; i <= n; i++)
  {
    for (int j = 0; j <= 2; j++) // sum[i-1]%3
    {
      for (int k = 0; k <= 2; k++) //相当于weight[i-1]%3;
      {
        int s = (j+ a[i]) % 3;
        int w = (j * a[i] + k) % 3;
        dp[i][s][w]+= dp[i - 1][j][k];
        
      }
    }
  }
  int ans = 0;
  for (int i = 1; i <= n; i++)
  {
    for (int k = 0; k <= 2; k++)
    {
      ans += dp[i][k][0];
    }
  }
  cout << ans << endl;
  return 0;
}
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值