首页 > 代码库 > Codeforces 413B Spyke Chatting(数论简单)

Codeforces 413B Spyke Chatting(数论简单)

题目链接:Codeforces 413B Spyke Chatting


题目大意:n个人,m种聊天器,k次发送消息,然后给出n*m的矩阵,如果g[i][j]为1,则表示i号人会使用j号聊天器,接着给出k次消息发送者和聊天器,如果i在j种聊天器上发送了一条消息,那么所有使用j种聊天器的人都会接受到消息。现在要求每个人会接受到几条消息,自己发送的不算。


解题思路:分别记录每个聊天器上有多少个消息,以及每个人发送了多少条消息,然后计算每个人接受到多少条消息的时候只要将这个人所使用的各个聊天器消息数取和在减去自己发送消息的个数即可。


#include <cstdio>
#include <cstring>

const int N = 2 * 1e4 + 5;

int n, m, k, c[N], r[15];
int g[N][15];

void init () {
	memset(r, 0, sizeof(r));
	memset(c, 0, sizeof(c));
	memset(g, 0, sizeof(g));

	scanf("%d%d%d", &n, &m, &k);

	for (int i = 1; i <= n; i++) {
		for (int j = 1; j <= m; j++)
			scanf("%d", &g[i][j]);
	}
}

void solve () {
	int a, b;
	for (int i = 0; i < k; i++) {
		scanf("%d%d", &a, &b);
		c[a]++;
		r[b]++;
	}

	for (int i = 1; i <= n; i++) {
		int t = 0;
		for (int j = 1; j <= m; j++) {
			if (g[i][j])
				t += r[j];
		}
		c[i] = t - c[i];
	}

	for (int i = 1; i < n; i++)
		printf("%d ", c[i]);
	printf("%d\n", c[n]);
}

int main () {
	init ();
	solve ();
	return 0;
}