zoj 2619 KMP+高斯消元+概率求期望

这道题一下子最对挺高兴的，发现自己最近敲代码之前总是发愣，很没效率啊！！！

用a[i]代表状态：尾部满足所给字符串长度i时为达到满足的字符串还所需要的期望。据此建立多元方程组然后进行高斯消元求解！

/
// File Name: 13614.cpp
// Author: wang
// mail: 
// Created Time: 2013-8-7 17:11:48
/
#include <cstdio>
#include <cstdlib>
#include <climits>
#include <cstring>
#include <cmath>

#include <algorithm>
#include<iostream>
#include<queue>
#include <map>
using namespace std;
typedef long long ll;
#define INF (INT_MAX/10)
#define SQR(x) ((x)*(x))
#define rep(i, n) for (int i=0; i<(n); ++i)
#define repf(i, a, b) for (int i=(a); i<=(b); ++i)
#define repd(i, a, b) for (int i=(a); i>=(b); --i)
#define clr(ar,val) memset(ar, val, sizeof(ar))
#define pb(i) push_back(i)
#define N 15

int n,m;
char s[N];
ll a[N][N];
int next[N];
bool vis[30];

void kmp()
{
	memset(next,-1,sizeof(next));
	int j=-1;
	int len=strlen(s);
	rep(i,len)
	{
		while(j!=-1 && s[i]!=s[j] ) j=next[j];
		next[i+1]=++j;
	}
	next[0]=0;
}
void solve()
{
    kmp();	//是为了进行匹配的
	memset(a,0,sizeof(a));
	memset(vis,false,sizeof(vis));
	vector<int>vec;
	int len=strlen(s);
	rep(i,len)
		if(vis[s[i]-'A']==false) vec.pb(s[i]),vis[s[i]-'A']=true;
	int mm=vec.size();
	rep(i,len)
	{
		a[i][i]-=n;//都扩大n倍的
		a[i][len+1]=n;
		a[i][0]+=n-mm;//多的那个会为0的
		//还有的就是包含的分开讨论
		rep(j,mm)
		{
			int x=vec[j]; 
			if(x==s[i]) a[i][i+1]++;
			else
			{
			  int k=next[i];
			  while(k!=0 && s[k]!=x) k=next[k];
			  if(s[k]==x) a[i][k+1]++;
			  else a[i][0]++;
			}
		}
	}
//	rep(i,len)
//	{
//		rep(j,len+2) cout<<a[i][j]<<" ";cout<<endl;
//	}
}
void guess()
{ 
	rep(i,n)
	{
		int k=-1;
		repf(j,i,n-1) 
			if(a[j][i]!=0)
			{
				k=j;break;
			}
		if(k==-1) continue;//这一列都为空的
		if(k!=i) 
			rep(j,m) swap(a[i][j],a[k][j]);
		rep(j,n)
		{
			if(j!=i)
			{
				int x=a[i][i],y=a[j][i];
				rep(k,m)
					a[j][k]=a[j][k]*x-a[i][k]*y;
			}
		}
	}
//	cout<<endl;
//	rep(i,n)
//	{
//		rep(j,m) cout<<a[i][j]<<" ";cout<<endl;
//	}
	if(a[0][0]==0) cout<<0<<endl;
	cout<<-a[0][m-1]/a[0][0]<<endl;
}
int main()
{
	int test;
	scanf("%d",&test);
	repf(i,1,test)
	{
		if(i!=1) printf("\n");
		printf("Case %d:\n",i);
		scanf("%d",&n);
		scanf("%s",s);
		m=strlen(s)+2;
		solve();
		n=strlen(s);        //高斯消元，n行len+2列的
		guess();
	}
	return 0;
}