[2019 ICPC Asia-East Continent Final] M. Value 状压+搜索

题目链接：M. Value

题解

本题如果直接暴力肯定会T。
但是如果将整个序列分组再搜索，那么时间复杂度就会大大减少。
按 $x^k$ (k≥1) 分为一组
例如 2，3，5，6，7，10。。作为每一组的开头，组内的元素就是开头的k次幂。

很显然每一组的大小最多不超过$lo{g}_2^{1e9}$，暴力搜索每组每个元素是否被选取即可。

代码

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<bitset>
#include<cassert>
#include<cctype>
#include<cmath>
#include<cstdlib>
#include<ctime>
#include<deque>
#include<iomanip>
#include<list>
#include<map>
#include<queue>
#include<set>
#include<stack>
#include<vector>
#include<unordered_set>
#include<unordered_map>
using namespace std;
//extern "C"{void *__dso_handle=0;}
typedef long long ll;
typedef long double ld;
typedef unsigned long long ull;
#define fi first
#define se second
#define pb push_back
#define mp make_pair
#define pii pair<int,int>
#define lowbit(x) x&-x

const double PI=acos(-1.0);
const double eps=1e-6;
const ll mod=1e9+7;
const int inf=0x3f3f3f3f;
const int maxn=1e5+100;
const int maxm=100+10;
#define ios ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);

ll a[maxn],b[maxn],dp[maxn][2],cnt[35];
vector<int> p,v[maxn];
int vis[maxn],n;

ll dfs(int pos,int k)
{
	if(pos>=v[k].size()) return 0;
	int tp=1,tmp=0;
	for(int i=1;;i++)
	{
		tp*=k;
		if(tp==v[k][pos])
		{
			tmp=i;
			break;
		}
	}
	for(int j=pos+tmp;j<v[k].size();j+=tmp) cnt[j]++;
	ll ans=0,ans1=0;
	ans=a[v[k][pos]]-cnt[pos]*b[v[k][pos]];
	ans+=dfs(pos+1, k);
	for(int j=pos+tmp;j<v[k].size();j+=tmp) cnt[j]--;
	ans1+=dfs(pos+1, k);
	return max(ans1,ans);
}

int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
		scanf("%lld",&a[i]);
	for(int j=1;j<=n;j++)
		scanf("%lld",&b[j]);
	for(ll i=2;i<=n;i++)
	{
		if(vis[i]) continue;
		vis[i]=0; p.push_back(i);
		for(ll j=i;j<=n;j*=i)
		{
			vis[j]=1;
			v[i].push_back(j);
		}
	}
	ll ans=a[1];
	for(int i=0;i<p.size();i++)
	{
		memset(cnt, 0, sizeof(cnt));
		ans+=dfs(0,p[i]);
	}
	printf("%lld\n",ans);
}