hdu 4039 The Social Network

hdu 4039 The Social Network

题意：给出N对好友关系，之后Q次提问，问可以对该用户推荐的相识度最高的好友；

推荐好友满足的条件：该用户的所有好友的好友中，出现次数最多的，而且推荐好友本身不是该用户的好友；若有多个推荐好友时，按字典序输出；

分析：

嘿嘿，我的做法有点暴力，600+ms；

首先对所有用户的名字（字符串）标号，好处理一些，这里用了字典树实现了；

接下来，用邻接矩阵表示任意俩个用户直接的关系，再者，对与每一次询问，遍历该用户所有好友的好友，找出出现次数最多的；

最后，若有多个推荐好友时，要字典树输出，额，我用优先队列做了，重载操作符，之后直接丢进去，再输出就可以了

#include<iostream>
#include<string>
#include<queue>
using namespace std;
int num;
short map[2001][2001];
struct str
{
    char s[16];
    friend bool operator<(const str ne1,const str ne2)//重载操作符
    {
        if(strcmp(ne1.s,ne2.s)==1)
            return true;
         return false;
    }
}st[2001];
typedef struct node  
{  
    int cnt;  
    struct node *next[26]; 
}*tree,Trie;  
tree root; 
int ex[2001],f[2001];
inline int GetNum(char *t)//用字典树对字符串编号
{
    tree p = root,newnode;
    for(int i = 0;i < strlen(t); ++i){
        int u = t[i] - 'a';
        if(p->next[u]==NULL)
        {    
            newnode=(tree)malloc(sizeof(Trie));
            newnode->cnt=-1;
            for(int j=0;j<26;j++)
                newnode->next[j]=NULL;
            p->next[u]=newnode;
            p=newnode;
        }
        else
        p = p->next[u];
    }
    if(p->cnt == -1) //该节点未出现过
        p->cnt = num ++;
    return p->cnt;
}
int main()
{
    int T=0,cas,n,q,n1,n2;
    char str1[16],str2[16];
    scanf("%d",&cas);
    while(cas--)
    {    
        root=(tree)malloc(sizeof(Trie));
        root->cnt=-1;
        for(int j=0;j<26;j++)
          root->next[j]=NULL;
        num=0;
        scanf("%d %d",&n,&q);
        memset(map,0,sizeof(map));
        for(int i=0;i<n;i++)
        {
            scanf("%s %s",str1,str2);
            n1=GetNum(str1);
            n2=GetNum(str2);
            strcpy(st[n1].s,str1);
            strcpy(st[n2].s,str2);
            map[n1][n2]=map[n2][n1]=1;
        }
        printf("Case %d:\n",++T);
        while(q--)
        {
            int max=0,count=0;
            memset(f,0,sizeof(f));
            memset(ex,0,sizeof(ex));
            scanf("%s",str1);
            n1=GetNum(str1);
            for(int i=0;i<num;i++)
            {
                if(n1==i)continue;//自己跟自己，肯定……
                if(map[n1][i]==1)
                {
                    for(int j=0;j<num;j++)
                    {
                        if(i==j||j==n1)continue;//推荐好友本身不能是自己的好友
                        if(!map[n1][j]&&map[i][j]==1)
                        {
                            ex[j]++;
                            if(ex[j]>=max)
                            {
                                if(ex[j]==max)//若有出现次数相同的，累加
                                    f[count++]=j;
                                else {
                                    count=0;//若有出现次数更多的，直接覆盖
                                    f[count++]=j;
                                }
                                max=ex[j];
                            }
                        }
                    }
                }
            }
            priority_queue<str> Q;
            if(count==0) printf("-");
            else
            {
                for(int i=0;i<count;i++)
                    Q.push(st[f[i]]);
            }
            for(int i=0;i<count;i++)
            {
                if(i==0)
                printf("%s",Q.top().s);
                else 
                printf(" %s",Q.top().s);
                Q.pop();
            }
            printf("\n");
       }
    }
        return 0;
}