【BZOJ 1097】 [POI2007]旅游景点atr

来源：互联网发布：用js写10秒倒计时编辑：程序博客网时间：2024/04/21 00:11

由于给出的第二个条件所以一定是要知道k+1个点之间的相互最短距离，但是Floyd肯定是没法搞的，观察k很小，直接暴力跑k次dijkstra就好了。然后设f[i][pos]表示到达i节点，经过节点状态为pos的最短距离，枚举状态然后枚举点转移就好了。

#include<cstdio>#include<cstring>#include<iostream>#include<queue>#define LL long long#define inf 0x3f3f3f3f#define maxn 20021#define maxm 200021#define MK make_pairusing namespace std;typedef pair<int,int> pii;int n,m,k,d[maxn],head[maxn],tot=1,f[22][1048576],vis[maxn],dis[23][23],pre[23];void read(int& x){x=0;char c=getchar();int f=1;for(;c>'9'||c<'0';c=getchar());if(c=='-')f=-1;for(;c>='0'&&c<='9';c=getchar())x=x*10+c-'0';x*=f;}struct edge{int v,w,next;}e[maxm*2];void adde(int a,int b,int c){e[tot].v=b,e[tot].w=c,e[tot].next=head[a];head[a]=tot++;}void dijkstra(int x){priority_queue<pii,vector<pii>,greater<pii> >q;for(int i=0;i<=n;i++)d[i]=inf,vis[i]=0;d[x]=0;q.push(MK(0,x));while(!q.empty()){int u=q.top().second;q.pop();if(vis[u])continue;vis[u]=1;for(int v,i=head[u];i;i=e[i].next){if(d[u]+e[i].w<d[v=e[i].v]){d[v]=d[u]+e[i].w;q.push(MK(d[v],v));}}}for(int i=1;i<=k+1;i++)dis[x][i]=d[i];dis[x][0]=d[n];}void solve(){f[1][0]=0;int t=(1<<k)-1;for(int i=0;i<=t;i++){for(int a=1;a<=k+1;a++)if(f[a][i]!=-1){for(int b=1;b<=k+1;b++)if((i&pre[b])==pre[b]){if(f[b][i|(1<<b-2)]==-1||f[a][i]+dis[a][b]<f[b][i|(1<<b-2)])f[b][i|(1<<b-2)]=f[a][i]+dis[a][b];}}}int ans=inf;for(int i=1;i<=k+1;i++)if(f[i][t]!=-1)ans=min(ans,f[i][t]+dis[i][0]);printf("%d",ans);}int main(){memset(f,-1,sizeof(f));read(n),read(m),read(k);for(int a,b,c,i=1;i<=m;i++){read(a),read(b),read(c);adde(a,b,c),adde(b,a,c);}for(int i=1;i<=k+1;i++)dijkstra(i);int q;read(q);for(int a,b,i=1;i<=q;i++){read(a),read(b);pre[b]|=1<<a-2;}solve();return 0;}

0 0