poj 2949 Word Rings (spfa+栈优化)

poj2949 题目链接 题意:我们有 n 个 (n<=100000) 字符串,每个字符串都是由 a~z 的小写英文字 母组成的字符串。如果字符串 A 的结尾两个字符刚好与字符串 B 的开头两字符相 匹配,那么我们称 A 与 B 能相连(注意: A 能与 B 相连不代表 B 能与 A 相连)。 我们希望从给定的字符串中找出一些,使得他们首尾相接形成一个环串(一个串首尾相连也算) 我们想要使这个环串的平均长度最长。

思路:参考了国家集训队论文《spfa算法的优化与应用》

首先我卡在了关于接龙问题的处理方法,只能想到n^2的方法。。显然gg.

而正解是把每个单词看做一条边,把每个单词开头的两个字母和结尾两个字母看做起点和终点,由于都是小写字母,2位26进制数最多表示26*26。

这个建图方式并不是特别显然,不过想一下还是可以理解的。。以及这应该算是处理单词接龙问题的一个技巧。。。

这道题综合了两种常见的问题:字符串的接龙以及平均值的最优化问题。对于前者,我们可以采取把单词看成边,把首尾字母组合看成点的方法。例如对于单词ababc就是点”ab”向点”bc”连一条长度为5的边。这样问题的模型变得更加清晰,规模也得到减小。那么原问题就可以转化成在此图中找一个环,使得环上边权的平均值最大。对于这种问题,我们有很经典的解决方法:

由于Average=(E1+E2+…..+Ek)/K 所以Average*K=E1+E2+……+Ek 即(E1-Average)+(E2-Average)+….+ (Ek-Average)=0 另外注意到上式中的等于号可以改写为小于等于,那么我们可以二分答案Ans,然后判断是否存在一组解满足(E1+E2+…..+Ek)/K>Ans,即判断 (E1- Ans)+(E2- Ans)+….+ (Ek- Ans)>0 于是在二分答案后,我们把边的权值更新,问题就变成了查找图中是否存在一个正环。

然后参考了这篇题解学习了一下栈优化的spfa: spfa栈优化   

以及这篇博客中比较了dfs的spfa和普通栈优化的spfa… 200+ms vs 2000+ms…十倍的优化。。太神了。。。

/* 枚举每一个平均长度mid值。 如果存在一个环,(E1+...+Ek)/k>=mid(其中k是边数,E1……Ek是各个边权), 那么正解比mid大,否则比mid小,这就是二分策略。 那么怎样知道是否存在(E1+...+Ek)/k>=mid 呢? 如下转化:(E1+...+Ek)>=mid*k E1-mid + E2–mid + E3-mid + ... + Ek-mid >= 0 所以,把所有的边权改为Ei – mid,然后看是否存在正环就可以,存在就是满足条件。 */
  1/* ***********************************************
  2Author :111qqz
  3Created Time :2016年05月25日 星期三 00时25分51秒
  4File Name :code/poj/2949.cpp
  5************************************************ */
  6
  7#include <cstdio>
  8#include <cstring>
  9#include <iostream>
 10#include <algorithm>
 11#include <vector>
 12#include <queue>
 13#include <set>
 14#include <map>
 15#include <string>
 16#include <cmath>
 17#include <cstdlib>
 18#include <ctime>
 19#include <stack>
 20#define fst first
 21#define sec second
 22#define lson l,m,rt<<1
 23#define rson m+1,r,rt<<1|1
 24#define ms(a,x) memset(a,x,sizeof(a))
 25typedef long long LL;
 26#define pi pair < int ,int >
 27#define MP make_pair
 28
 29using namespace std;
 30const double eps = 1E-8;
 31const int dx4[4]={1,0,0,-1};
 32const int dy4[4]={0,-1,1,0};
 33const int inf = 0x3f3f3f3f;
 34const int N=1E5+7;
 35char st[1005];
 36int f[900];
 37bool ins[900];//是否在栈中
 38double dis[900];
 39int in[900]; //进栈次数,用来判断环。
 40double maxedge; // d[i]>n*maxegde 可以认定必然有环。 这剪枝真的有效果。。?
 41
 42struct edge
 43{
 44    int v,nxt;
 45    double w;
 46}edge[N];
 47int n;
 48int tot;
 49int cnt;//点的个数。
 50int head[N];
 51
 52void addedge(int u,int v,double w)
 53{
 54    edge[tot].v = v;
 55    edge[tot].w = w;
 56    edge[tot].nxt = head[u];
 57    head[u] = tot++;
 58}
 59
 60bool spfa(double ave,int n)
 61{
 62
 63    ms(in,0);
 64    ms(ins,false);
 65    ms(dis,0);
 66    stack<int>S;
 67    for ( int i = 1 ; i <= n ; i++) S.push(i),ins[i] = true;
 68//    cout<<"whhhhh?"<<endl; 
 69    while (!S.empty())
 70    {
 71	int u = S.top();
 72//	cout<<"u:"<<u<<endl;
 73	S.pop();
 74	ins[u]=false;
 75
 76	for ( int i = head[u] ; i!=-1 ; i = edge[i].nxt)
 77	{
 78//	    cout<<"end of for?"<<endl;
 79	    int v = edge[i].v;
 80	    if (dis[v]<dis[u]+(edge[i].w-ave))
 81	    {
 82		dis[v] = dis[u] +  (edge[i].w-ave);
 83		if (dis[v]>maxedge*n) return true;
 84		if (ins[v]) continue;
 85		ins[v] = true;
 86		S.push(v);
 87		in[v]++;
 88		if (in[v]>n)
 89		{
 90		    return true; //有环。
 91		}
 92	    }
 93	}
 94    }
 95
 96  //  cout<<"end of spfa:"<<endl;
 97
 98    return false;
 99
100}
101
102void bin()
103{
104    double l = 0,r=1000,best=-1,mid;
105    while (l+0.001<=r)
106    {
107//	cout<<"sadasd"<<endl;
108	mid = (l+r)/2;
109	if (spfa(mid,cnt))
110	{
111	    best = mid;
112	    l = mid;
113	}
114	else r = mid;
115    }
116    if (best!=-1) printf("%.2lf\n",best);
117    else puts("No solution.");
118}
119int main()
120{
121	#ifndef  ONLINE_JUDGE 
122	freopen("code/in.txt","r",stdin);
123  #endif
124
125	while (scanf("%d",&n)!=EOF)
126	{
127	    if (n==0) break;
128	    ms(f,0);
129	    ms(head,-1); //总是忘记这个初始化....
130	    cnt = 0;
131	    tot = 0 ;
132	    for ( int i = 1 ; i <= n ; i++)
133	    {
134		scanf("%s",st);
135		int len =  strlen(st);
136		if (len>maxedge)
137		{
138		    maxedge = len;
139		}
140
141		int x = (st[0]-'a')*26 + st[1]-'a';
142		int y = (st[len-2]-'a')*26 + st[len-1]-'a';
143		if (f[x]==0) f[x] = ++cnt;
144		if (f[y]==0) f[y] = ++cnt;
145
146		int u = f[x];
147		int v = f[y];
148
149		addedge(u,v,double(len));
150	    }
151
152
153	   bin();
154	//    cout<<"asdasd"<<endl;
155	}
156
157
158
159  #ifndef ONLINE_JUDGE  
160  fclose(stdin);
161  #endif
162    return 0;
163}