bnuoj 20838 Item-Based Recommendation （模拟）

http://www.bnuoj.com/bnuoj/problem_show.php?pid=20838

【题意】：

　　有点长，略。

【code】：

　　

#include <iostream>
#include <stdio.h>
#include <string>
#include <string.h>
#include <math.h>
#include <algorithm>

using namespace std;

double rat[220][220];
double drat[220][220];

struct Nod
{
    int id;
    double rt;
}node[220];

int m,n;

bool cmp(Nod a,Nod b)
{
    return a.rt>b.rt;
}

double getAns(int a,int b)
{
    int i;
    double sum=0;
    for(i=1;i<=n;i++)
    {
        if(rat[i][a]>0&&rat[i][b]>0)
        {
            sum+=(rat[i][a]-rat[i][b])*(rat[i][a]-rat[i][b]);
        }
    }
    return 1.0/(sum+1);
}

void getDrat()
{
    int i,j;
    for(i=1;i<=m;i++)
    {
        drat[i][i]=1;
        for(j=i+1;j<=m;j++)
        {
            drat[i][j]=drat[j][i]=getAns(i,j);
        }
    }
}

int main()
{
    int c;
    scanf("%d%d%d",&n,&m,&c);
    memset(rat,0,sizeof(rat));
    while(c--)
    {
        int a,b;
        scanf("%d%d",&a,&b);
        scanf("%lf",&rat[a][b]);
    }
    getDrat();
    int x;
    while(~scanf("%d",&x))
    {
        printf("Recommendations for user %d:
",x);
        int i,cnt=0,j;
        double down=0,up=0;
        for(i=1;i<=m;i++)
        {
            if(rat[x][i]==0)//i=6
            {
                down=0,up=0;
                for(j=1;j<=m;j++)
                {
                    if(rat[x][j]>0)
                    {
                        down+=drat[i][j];
                        up+=drat[i][j]*rat[x][j];
                    }
                }
                node[cnt].id = i;
                node[cnt].rt = up/down;
                cnt++;
            }
        }
        sort(node,node+cnt,cmp);
        for(i=0;i<cnt&&i<10;i++)
        {
            printf("%d %.3lf
",node[i].id,node[i].rt);
        }
        putchar(10);
    }
    return 0;
}