..k次最短路后,考虑如何满足先走一些点

用状压dp,每一个点考虑它所需要经过的点a[i],当当前走过的点包含a[i]时,i 这个点才可以到达。

写的时候用记忆化搜索。

 1 #include<bits/stdc++.h>
 2 #define ll long long
 3 using namespace std;
 4 inline int read(){
 5   int x=0,f=1;char ch=getchar();
 6   while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
 7   while(ch>='0'&&ch<='9'){x=10*x+ch-'0';ch=getchar();}
 8   return x*f;
 9 }
10 #define M 200005 
11 #define N 20005
12 #define inf 1e9
13 struct Node{
14   int to,next,v;
15 }e[M<<1];
16 int n,m,k,tot,head[N],d[25][N],a[25],dp[25][2100000];
17 bool vis[N];
18 void add(int x,int y,int z){
19   e[++tot]=(Node){y,head[x],z};head[x]=tot;
20   e[++tot]=(Node){x,head[y],z};head[y]=tot;
21 }
22 queue<int> q;
23 void spfa(int x){
24   for(int i=1;i<=n;i++)d[x][i]=inf;
25   d[x][x]=0;vis[x]=1;q.push(x);
26   while(!q.empty()){
27     int now=q.front();q.pop();vis[now]=0;
28     for(int i=head[now];i;i=e[i].next)
29       if(d[x][now]+e[i].v<d[x][e[i].to]){
30         d[x][e[i].to]=d[x][now]+e[i].v;
31         if(!vis[e[i].to]){
32           vis[e[i].to]=1;q.push(e[i].to);
33         }
34       }
35   }
36 }
37 int calc(int x){
38   int c=0;
39   while(x){
40     if(x&1)c++;
41     x>>=1;
42   }
43   return c;
44 }
45 int dfs(int x,int y){
46   if(dp[x][y]>=0)return dp[x][y];
47   if(y==(1<<k)-1)return d[x+1][n];
48   dp[x][y]=inf;
49   for(int i=1;i<=k;i++)
50     if((y|a[i])==y)
51       dp[x][y]=min(dp[x][y],d[x+1][i+1]+dfs(i,y|(1<<(i-1))) );
52   return dp[x][y];
53 }
54 int main(){
55  // freopen("1097.in","r",stdin);
56   n=read();m=read();k=read();
57   for(int i=1;i<=m;i++){
58     int x=read(),y=read(),z=read();
59     add(x,y,z);
60   }
61   for(int i=1;i<=k+1;i++)spfa(i);
62   memset(dp,-1,sizeof(dp));
63   int t=read();
64   while(t--){
65     int x=read()-1,y=read()-1;
66     a[y]|=1<<(x-1);
67   }
68   printf("%d\n",dfs(0,0));
69   return 0;
70 }
View Code

相关文章: