ZOJ3560 Re:the Princess(高斯消元法)

题目要读很久才能理解它的意思和笑点(如果你也看过那个笑话的话),读懂之后就会发现是一个高斯消元法的题目,对于我来说难点不在高斯消元,而在于字符串处理。先来说说题意吧:

总共有n个人,n个人都会有一段话,先是princess说话,里面如果提到了a1,a2,a3...这几个不同的人的话,对应提到的次数是x1,x2,x3..的话,那么下一个对话是ai这个人说的概率是xi/(x1+x2+x3)....,然后下一个人的对话里也会提到别的人,然后也有一定的概率会有下一轮对话,现在要问的就是,给定了这些对话,问你期望的对话次数是多少。

我们可以设第i个人持续的对话的期望是xi,那么xi应该等于  xi=p1*x1+p2*x2+p3*x3+...+1 对应的pi即为在该人对话出现的次数所对应的频率。然后对每个人列出这样的方程就会构成一系列的方程组。然后高斯消元即可。

难度在于算在某个对话里出现了多少次,因为像如何你用KMP去做匹配 prince是会出现在princess那里的,题目说了人名之间是用五种间隔符隔开的,所以匹配的时候或者匹配前将对话里的每一个词分出来,然后和对应的字符串判相等即可。我写的不好跪了好几发呀- -0

#pragma warning(disable:4996)
#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<vector>
#define ll long long
#define maxn 220
#define eps 1e-7
using namespace std;

char str[120][1050];
char name[120][20];
int n;

double mat[120][120];
char word[1200];

int dcmp(double x) {
	return (x > eps) - (x < -eps);
}

int cnt(char*T, char *S)
{
	int res = 0; int idx = 0;
	for (int i = 0; T[i]; i++){
		if (!(T[i] <= 'z'&&T[i] >= 'a')) continue;
		int id = 0;
		while (T[i] <= 'z'&&T[i] >= 'a'){
			word[id++] = T[i++];
		}
		word[id] = '\0';
		if (strcmp(word, S) == 0) res++;
		i--;
	}
	return res;
}

bool gauss(double mat[120][120], int n)
{
	for (int i = 0; i < n; i++){
		int pivot = i;
		for (int j = i; j < n; j++){
			if (abs(mat[j][i])>abs(mat[pivot][i])) pivot = j;
		}
		swap(mat[i], mat[pivot]);
		if (abs(mat[i][i]) < eps) return false;
		for (int j = i + 1; j <= n; j++) mat[i][j] /= mat[i][i];
		for (int j = 0; j < n; j++){
			if (i != j)
			for (int k = i + 1; k <= n; k++){
				mat[j][k] -= mat[j][i] * mat[i][k];
			}
		}
	}
	return true;
}


int main()
{
	while (cin >> n)
	{
		getchar();
		memset(name, 0, sizeof(name));
		memset(str, 0, sizeof(str));
		for (int i = 0; i < n; i++){
			gets(str[i]);
			int j;
			for (j = 0; str[i][j]; j++){
				if (str[i][j] == ':') break;
				name[i][j] = str[i][j];
			}
			name[i][j + 1] = '\0'; int k = 0; j++;
			for (; str[i][j]; j++,k++){
				str[i][k] = str[i][j];
			}
			str[i][k] = '\0';
		}
		memset(mat, 0, sizeof(mat));
		for (int i = 0; i < n; i++){
			double tot = 0; double tmp = 0;
			for (int j = 0; j < n; j++){
				if (i == j) continue;
				tmp = cnt(str[i], name[j]);
				mat[i][j] = tmp;
				tot += tmp;
			}
			mat[i][i] = -tot; mat[i][n] = -tot;
			if (dcmp(tot) == 0){
				mat[i][i] = mat[i][n] = -1;
			}
		}
		if (gauss(mat, n)){
			printf("%.3lf\n", mat[0][n]);
		}
		else puts("Infinity");
	}
	return 0;
}

 

posted @ 2014-03-31 19:48  chanme  阅读(270)  评论(0编辑  收藏  举报