2019.7.14 义乌模拟赛 T3 set

你会发现这个东西看上去很玄学。
然后这个你会发现其实\(2\)和\(3\)是互不影响的。
所以我们对于每个数的后继都找到然后状压一下看看最大值然后最后将所有加起来即可。
注意如果一个数被算过了那么打上标记不再算。
时间复杂度\(O(能过)\)
code:

#include<bits/stdc++.h>
#define I inline
#define max(a,b) ((a)>(b)?(a):(b))
#define min(a,b) ((a)<(b)?(a):(b))
#define abs(x) ((x)>0?(x):-(x))
#define re register
#define ll long long
#define db double
#define N 100000
#define M 1500
#define mod 1000000000
#define mod2 39989
#define eps (1e-7)
#define U unsigned int
#define it iterator
#define Gc() getchar() 
#define Me(x,y) memset(x,y,sizeof(x))
using namespace std;
int n,m,k,A[N+5],B[N+5],fl[N+5],C[N+5],cnt;ll ans,tot,pus,now;
I void calc(int x,int y){
	re ll i,j;tot=1;for(i=0;i<cnt;i++) tot*=x,C[tot]=(y>>i)&1;now=0;
	for(i=x;i<=n;i*=x){
		if(C[i]) now+=A[i];tot=1ll*i*i;while(tot<=n) now-=C[i]*C[tot]*B[tot],tot*=i; 
	}pus=max(pus,now);tot=1;for(i=0;i<cnt;i++) tot*=x,C[tot]=0;
}
int main(){
	freopen("set.in","r",stdin);freopen("set.out","w",stdout);
	re int i,j,h;scanf("%d",&n);for(i=1;i<=n;i++) scanf("%d",&A[i]);for(i=1;i<=n;i++) scanf("%d",&B[i]);ans=A[1];
	for(i=2;i<=n;i++) {//printf("%d\n",i);
		if(fl[i]) continue;cnt=0;tot=i;while(tot<=n)cnt++,fl[tot]=1,tot*=i;pus=0;
		for(j=1;j<(1<<cnt);j++)calc(i,j);
		ans+=pus; 
	}printf("%lld\n",ans);
}
上一篇:洛谷P1352—没有上司的舞会(树形DP)


下一篇:做题记录 Luogu P1352