先上题解吧
题解讲的算是比较详细了,用状压DP写。但是第二类公司的集合形成方式,没有详细的讲明,应该要用离散化的方式将其压缩空间,不然会爆空间。(用num的vector容器进行离散)
还有一点就是,时间题目卡了时间复杂度。单纯的用深搜跑图会超时掉,所以要进行图优化。图优化是我在网上找题解找到的一个图优化。就是如果1->2,2->3,1->3,在这样的图中,1->3这条路就没有跑的必要了,因为跑的结果一定比1->2,2->3这样的方式要小,所以把这条路删掉,做一个剪枝。虽说是剪枝但在DFS中却大大降低了时间复杂度。(用prime数组进行剪枝,剪完枝,剪完后再把边存入vec变量vector容器中)
#include <bits/stdc++.h>
using namespace std;
const int maxn=1<<21;
int n,m;
int dp[38][maxn];
int c[40],w[40],sum[40];
int flag[40],prime[40][40];
vector<int> vec[40],num;
void DFS(int pre,int pos,int state)
{
int t=lower_bound(num.begin(),num.end(),c[pos])-num.begin();
if(flag[c[pos]]>=2&&(state&(1<<t))==0) dp[pos][state|(1<<t)]=dp[pre][state]+w[c[pos]];
else if(flag[c[pos]]>=2) dp[pos][state]=dp[pre][state];
else if(flag[c[pos]]==1) dp[pos][state]=dp[pre][state]+w[c[pos]];
if(flag[c[pos]]>=2) sum[pos]=max(sum[pos],dp[pos][state|(1<<t)]);
else sum[pos]=max(sum[pos],dp[pos][state]);
for(int i=0;i<(int)vec[pos].size();i++)
{
int v=vec[pos][i];
if(v<=pos||v>n) continue;
if(flag[c[pos]]>=2) DFS(pos,v,state|(1<<t));
else DFS(pos,v,state);
}
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(0);cout.tie(0);
cin>>n>>m;
for(int i=1;i<=n;i++)
{
cin>>c[i];
if(flag[c[i]]) num.push_back(c[i]);
flag[c[i]]++;
}
if(num.size())
{
sort(num.begin(),num.end());
num.erase(unique(num.begin(),num.end()),num.end());
}
for(int i=1;i<=n;i++) cin>>w[i];
for(int i=1;i<=m;i++)
{
int x,y;
cin>>x>>y;
prime[x][y]=1;
}
for (int i=1;i<=n;i++)
{
for (int j=i+2;j<=n;j++)
{
for (int q=i+1;q<=j-1;q++)
{
if (prime[i][j]&&prime[q][j]&&prime[i][q])prime[i][j] = 0;
}
}
}
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
if(prime[i][j])
vec[i].push_back(j);
DFS(1,1,0);
for(int i=1;i<=n;i++)
for(int j=0;j<=(int)num.size()+1;j++)
sum[i]=max(sum[i],dp[i][1<<j]);
for(int i=1;i<=n;i++)
cout<<sum[i]<<endl;
return 0;
}