http://acm.hdu.edu.cn/showproblem.php?pid=4039
The 36th ACM/ICPC Asia Regional Chengdu Site —— Online Contest
1009
题意:模拟一个社交网络系统,先给你n个朋友关系,对每个人,系统会推荐给他一些人给他交朋友,系统推荐的是他朋友的朋友,而且只推荐他最多个朋友的朋友,如果会推荐多个,按照字典序从小到大输出,如果一个也不推荐输出“-”。。。1 ≤ N, Q ≤ 1000
分析:由于数据量小,直接暴力即可,找他所有的朋友的朋友只需要n^2即可,二重for循环搞定,为节省时间,用邻接表存,因为每个人是以名字形式给出的,用字典树做一个映射。。。注意一个小地方,就是a的朋友的朋友可能是a的朋友。。。
代码:其实就是一水题,只是rank排在前面了,贴出来玩玩。。
#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <cmath>
#include <vector>
#include <map>
#include <iostream>
using namespace std;
const int N=2010;
int n, q, up, num;
char s[20], s1[20];
struct trie
{
int a[26];
int num;
void init()
{
memset(a, -1, sizeof(a));
num = -1;
}
} t[10000];
vector<int> a[N];
vector<int>::iterator it, it1;
int cnt[N], flag[N], flag1;
struct node
{
int i;
char s[16];
} ss[N], ss1[N];
inline int insert(char *s)
{
int p=0;
while(*s)
{
if(t[p].a[*s-'a']==-1)
{
t[up].init();
t[p].a[*s-'a'] = up++;
}
p = t[p].a[*s-'a'];
s++;
}
if(t[p].num==-1)
{
t[p].num = num++;
flag1 = 1;
}
return t[p].num;
}
inline int query(char *s)
{
int p=0;
while(*s)
{
if(t[p].a[*s-'a']==-1)
return -1;
p = t[p].a[*s-'a'];
s++;
}
return t[p].num;
}
int cmp(const node &a, const node &b)
{
return strcmp(a.s, b.s)<0;
}
int main()
{
int i, j, cas, cas1, x, y, mx;
scanf("%d", &cas);
for(cas1=1; cas1<=cas; cas1++)
{
scanf("%d%d%d", &n, &q);
t[0].init();
up = num = 1;
for(i=1; i<=n; i++)
{
scanf("%s %s", s, s1);
flag1 = 0;
x = insert(s);
if(flag1==1)
strcpy(ss[x].s, s);
flag1 = 0;
y = insert(s1);
if(flag1==1)
strcpy(ss[y].s, s1);
a[x].push_back(y);
a[y].push_back(x);
}
printf("Case %d:\n", cas1);
while(q--)
{
scanf("%s", s);
for(i=1; i<num; i++)
cnt[i] = 0;
x = query(s);
mx = 0;
for(i=1; i<num; i++)
flag[i] = 0;
flag[x] = 1;
for(it=a[x].begin(); it!=a[x].end(); it++)
flag[*it] = 1;
for(it=a[x].begin(); it!=a[x].end(); it++)
{
y = *it;
for(it1=a[y].begin(); it1!=a[y].end(); it1++)
{
if(flag[*it1]==0)
{
cnt[*it1]++;
if(cnt[*it1]>mx)
mx = cnt[*it1];
}
}
}
if(mx==0)
{
printf("-\n");
continue;
}
up = 0;
for(i=1; i<num; i++)
{
if(cnt[i]==mx)
{
strcpy(ss1[up].s, ss[i].s);
up++;
}
}
sort(ss1, ss1+up, cmp);
printf("%s", ss1[0].s);
for(i=1; i<up; i++)
printf(" %s", ss1[i].s);
printf("\n");
}
for(i=1; i<num; i++)
while(!a[i].empty())
a[i].pop_back();
}
return 0;
}