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;

}

 

你可能感兴趣的:(ZOJ)