题目大意:
给你一个n
再给你n个数
问你有多少个区间中两两乘积之和是能被3整除的
思路:
问了一下七爷思路,dp
对于每个新加入的a[i]我们可以这样考虑
weight[i]=(weight[i-1]+(a[i]*sum)%3)%3
sum[i]=(sum[i-1]*a[i])%3
那么我们就可以设dp[i][weight][sum]表示状态
对于出现过的weight与sum都在这个状态保存下来
那么如果我们设k=sum[i-1],j=weight[i-1]
方程就是
dp[i][weight[i]][sum[i]]+=dp[i-1][j][k]
核心代码
for(int i=1;i<=n;i++){
for(int j=0;j<3;j++){// j表示weight[i-1]%3
for(int k=0;k<3;k++){// k表示sum[i-1]%3
int ans=(j+(a[i]*k)%3)%3;//weight[i]=(weight[i-1]+(a[i]*sum)%3)%3
int sum=(k+a[i])%3;//sum[i]=(sum[i-1]*a[i])%3
dp[i][ans][sum]+=dp[i-1][j][k];
}
}
}
这里可能会有疑惑
因为每个区间的sum与weight不是固定的吗
怎么能建循环呢
这里dp[i]存的是以i为R但是L不确定的所有的区间对符合的个数
这里L是不确定的
并且因为要整除3其实只有3*3=9种状态,直接把9种状态看一遍把前面状态对应的值给它加上就能得到这个状态的值了
AC代码:
#include <iostream>
using namespace std;
const int N=5e5+5;
int dp[N][3][3];
int a[N];
int main()
{
ios::sync_with_stdio(false);
int n;
cin>>n;
for(int i=1;i<=n;i++){
cin>>a[i];
dp[i][0][a[i]%3]++;
}
for(int i=1;i<=n;i++){
for(int j=0;j<3;j++){// j表示weight[i-1]%3
for(int k=0;k<3;k++){// k表示sum[i-1]%3
int ans=(j+(a[i]*k)%3)%3;//weight[i]=(weight[i-1]+(a[i]*sum)%3)%3
int sum=(k+a[i])%3;//sum[i]=(sum[i-1]*a[i])%3
dp[i][ans][sum]+=dp[i-1][j][k];
}
}
}
for(int i=1;i<=n;i++){
for(int j=0;j<3;j++){// j表示weight[i-1]%3
for(int k=0;k<3;k++){// k表示su
cout<<dp[i][j][k]<<' ';
}
}
cout<<endl;
}
long long ans=0;
for(int i=1;i<=n;i++){
for(int k=0;k<3;k++){
ans+=dp[i][0][k];//只需要weight=0的值
}
}
cout<<ans<<endl;
return 0;
}