Description&Data Constraint
从前有一位旅者,他想要游遍天下所有的景点。这一天他来到了一个神奇的王国:在这片土地上,有n个城市,从1到n进行编号。王国中有m条道路,第i条道路连接着两个城市ai,bi,由于年代久远,所有的道路都已经不能使用。如果要修复第i条道路,需要wi的时间。为了更好的旅行,旅者想要将某些道路修复,使得1号城市能够到达n号城市,2号城市能够到达n-1号城市…k号城市能够到达n-k+1号城市。为了满足他的要求,请问最少需要多少时间去修复道路。无解请输出-1。
$k\le4, 2k\le n\le10^4, m\le10^4, wi\le10^3, 1 \le a_i, b_i \le n $
Solution
正解是不完全的斯坦纳树板子题
但是我不会!准确来说是能理解但不会讲
所以我来讲水法(JZOJ可以过)
由于 k k k比较小,可以全排列
然后每个走最短路,走过的 v a l val val改为0,可以通过 p r e pre pre来实现
最后在全排列中花费最少的就是答案
当然是很容易被hack的
所以还是去学斯坦纳树吧!
Code
- 水法
#include<cstdio>
#include<cstring>
#define N 10005
#define inf 1234567890
using namespace std;
struct node
{
int head,next,to,val,flag;
}a[N*2],save[N*2];
int n,m,k,x,y,z,tot,ans,res,num,c[5],dis[N],d[N<<2],g[N];
bool B[5],b[N],bj[N],bz[N];
void add(int x,int y,int z,int Flag)
{
a[++tot].to=y;
a[tot].val=z;
a[tot].flag=Flag;
a[tot].next=a[x].head;
a[x].head=tot;
}
int spfa(int x,int y)
{
int h=0,t=1;
d[1]=x;g[x]=0;bj[x]=true;dis[x]=0;
while (h<t)
{
int u=d[++h];
for (int i=a[u].head;i;i=a[i].next)
{
int v=a[i].to;
if (dis[u]+a[i].val<dis[v])
{
dis[v]=dis[u]+a[i].val;
if (a[i].flag==1) g[v]=i+1;
else g[v]=i-1;
if (!bj[v]) d[++t]=v,bj[v]=true;
}
}
bj[u]=false;
}
return dis[y];
}
void dg(int x)
{
if (x>k)
{
res=0;
for (int i=1;i<=m*2;++i)
{
save[i].flag=a[i].flag;
save[i].head=a[i].head;
save[i].next=a[i].next;
save[i].to=a[i].to;
save[i].val=a[i].val;
}
for (int i=1;i<=k;++i)
{
memset(bj,false,sizeof(bj));
memset(d,0,sizeof(d));
memset(g,0,sizeof(g));
memset(dis,0X3f3f3f3f,sizeof(dis));
res+=spfa(c[i],n-c[i]+1);
int l=g[n-c[i]+1];
while (l)
{
a[l].val=a[l+a[l].flag].val=0;
l=g[a[l].to];
}
}
if (res<ans) ans=res;
for (int i=1;i<=m*2;++i)
{
a[i].flag=save[i].flag;
a[i].head=save[i].head;
a[i].next=save[i].next;
a[i].to=save[i].to;
a[i].val=save[i].val;
}
return;
}
for (int i=1;i<=k;++i)
{
if (!B[i])
{
c[x]=i;
B[i]=true;
dg(x+1);
B[i]=false;
c[x]=0;
}
}
}
int main()
{
scanf("%d%d%d",&n,&m,&k);
for (int i=1;i<=m;++i)
{
scanf("%d%d%d",&x,&y,&z);
add(x,y,z,1);add(y,x,z,-1);
}
ans=inf;
dg(1);
if (ans==inf) printf("-1\n");
else printf("%d\n",ans);
return 0;
}
- 正解
#include<queue>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define inf 0X3f3f3f3f
#define N 10005
using namespace std;
struct node
{
int to,next,head,val;
}a[N*2];
queue <int> q;
int n,m,k,x,y,z,tot,f[N][260],ans[260],res[260];
bool b[N];
void add(int x,int y,int z)
{
a[++tot].to=y;
a[tot].val=z;
a[tot].next=a[x].head;
a[x].head=tot;
}
void spfa(int s)
{
memset(b,true,sizeof(b));
while (!q.empty())
{
int u=q.front();
q.pop();
for (int i=a[u].head;i;i=a[i].next)
{
int v=a[i].to;
if (f[v][s]>f[u][s]+a[i].val)
{
f[v][s]=f[u][s]+a[i].val;
if (!b[v])
{
q.push(v);
b[v]=true;
}
}
}
b[u]=false;
}
}
bool pd(int x)
{
for (int i=1;i<=k;++i)
{
int b1=x&(1<<(i-1)),b2=x&(1<<(k+i-1));
if ((!b1&&b2)||(b1&&!b2)) return false;
}
return true;
}
int main()
{
memset(f,inf,sizeof(f));
memset(res,inf,sizeof(res));
memset(ans,inf,sizeof(ans));
scanf("%d%d%d",&n,&m,&k);
for (int i=1;i<=m;++i)
{
scanf("%d%d%d",&x,&y,&z);
add(x,y,z);add(y,x,z);
}
for (int i=1;i<=k;++i)
f[i][1<<(i-1)]=f[n-i+1][1<<(k+i-1)]=0;
for (int s=1;s<(1<<(2*k));++s)
{
for (int i=1;i<=n;++i)
{
for (int ss=s&(s-1);ss;ss=s&(ss-1))
f[i][s]=min(f[i][s],f[i][ss]+f[i][s^ss]);
q.push(i);
}
spfa(s);
for (int i=1;i<=n;++i)
res[s]=min(res[s],f[i][s]);
}
ans[0]=0;
for (int s=1;s<(1<<(2*k));++s)
{
if (!pd(s)) continue;
for (int ss=0;ss<s;++ss)
if ((ss|s)==s) ans[s]=min(ans[s],ans[ss]+res[s^ss]);
}
if (ans[(1<<(2*k))-1]>=inf) printf("-1\n");
else printf("%d\n",ans[(1<<(2*k))-1]);
return 0;
}