Submission #93814

#	Time	Username	Problem	Language	Result	Execution time	Memory
93814		Bodo171	Friend (IOI14_friend)	C++14	35 / 100	66 ms	12108 KiB

friend

#include "friend.h"
#include <vector>
#include <iostream>
// Find out best sample
using namespace std;
const int nmax=100005;
vector<int> v[nmax],ad[nmax],inloc[nmax];
int sum[nmax][2],best[nmax][2];
int a[nmax],val[nmax];
int nr,p,tip;
void dfs(int x)
{
    for(int i=0;i<v[x].size();i++)
       dfs(v[x][i]);
    for(int i=0;i<inloc[x].size();i++)
        dfs(inloc[x][i]);
    for(int i=0;i<ad[x].size();i++)
        dfs(ad[x][i]);
    nr=0;
    for(int i=0;i<v[x].size();i++)
    {
        a[++nr]=v[x][i];
        sum[nr][0]=sum[nr-1][0]+best[a[nr]][0];
        sum[nr][1]=sum[nr-1][1]+max(best[a[nr]][0],best[a[nr]][1]);
    }
    int totad=0,totinloc=0;
    for(int i=0;i<ad[x].size();i++)
        totad+=best[ad[x][i]][0];
    for(int i=0;i<inloc[x].size();i++)
        totinloc+=best[inloc[x][i]][0];
    best[x][1]=sum[nr][0]+val[x]+totad+totinloc;
    for(int i=0;i<ad[x].size();i++)
    {
        p=0;
        while(a[p]<=ad[x][i]&&p<=nr)
            p++;
        p--;
        best[x][1]=max(best[x][1],sum[p][0]+sum[nr][1]-sum[p][1]+best[ad[x][i]][1]+totad+totinloc-best[ad[x][i]][0]);
        best[x][1]=max(best[x][1],sum[nr][0]+val[x]+best[ad[x][i]][1]+totad+totinloc-best[ad[x][i]][0]);
        totad+=best[ad[x][i]][1]-best[ad[x][i]][0];
    }
    totad=0;
    for(int i=0;i<ad[x].size();i++)
        totad+=best[ad[x][i]][0];
    for(int i=0;i<inloc[x].size();i++)
    {
        p=0;
        while(a[p]<=inloc[x][i]&&p<=nr)
            p++;
        p--;
        best[x][1]=max(best[x][1],best[inloc[x][i]][1]+sum[nr][1]-sum[p][1]+sum[p][0]+totad+totinloc-best[inloc[x][i]][0]);
    }
    for(int i=0;i<v[x].size();i++)
        best[x][0]+=max(best[v[x][i]][0],best[v[x][i]][1]);
    for(int i=0;i<ad[x].size();i++)
        best[x][0]+=best[ad[x][i]][0];
    for(int i=0;i<inloc[x].size();i++)
        best[x][0]+=best[inloc[x][i]][0];

}
void add(int x,int y)
{
    ad[x][y]=ad[y][x]=1;
}
int findSample(int n,int confidence[],int host[],int protocol[]){
    for(int i=0;i<n;i++)
        val[i]=confidence[i];
    if(n<=10)
    {
        int j,k,ok,mx=0;
        for(int i=0;i<n;i++)
            for(j=0;j<n;j++)
              ad[i].push_back(0);
        for(int i=1; i<n; i++)
        {
            tip=protocol[i];
            if(tip==0||tip==2)
            {
                add(i,host[i]);
            }
            if(tip==1||tip==2)
            {
                for(j=0;j<n;j++)
                    if(ad[host[i]][j])
                        add(i,j);
            }
        }
        for(int i=0;i<(1<<n);i++)
        {
            ok=1;
            for(int j=0;j<n;j++)
                for(int k=0;k<n;k++)
                   if(((1<<j)&i)&&((1<<k)&i)&&ad[j][k])
                      ok=0;
            if(ok)
            {
                int act=0;
                for(int j=0;j<n;j++)
                    if(((1<<j)&i))
                       act+=confidence[j];
                mx=max(mx,act);
            }
        }
        return mx;
    }
    for(int i=1;i<n;i++)
    {
        tip=protocol[i];
        if(tip==0)
        {
            v[host[i]].push_back(i);
        }
        if(tip==1)
        {
            ad[host[i]].push_back(i);
        }
        if(tip==2)
        {
            inloc[host[i]].push_back(i);
        }
    }
    dfs(0);
	return max(best[0][0],best[0][1]);
}

Compilation message (stderr)

friend.cpp: In function 'void dfs(int)':
friend.cpp:13:18: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     for(int i=0;i<v[x].size();i++)
                 ~^~~~~~~~~~~~
friend.cpp:15:18: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     for(int i=0;i<inloc[x].size();i++)
                 ~^~~~~~~~~~~~~~~~
friend.cpp:17:18: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     for(int i=0;i<ad[x].size();i++)
                 ~^~~~~~~~~~~~~
friend.cpp:20:18: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     for(int i=0;i<v[x].size();i++)
                 ~^~~~~~~~~~~~
friend.cpp:27:18: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     for(int i=0;i<ad[x].size();i++)
                 ~^~~~~~~~~~~~~
friend.cpp:29:18: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     for(int i=0;i<inloc[x].size();i++)
                 ~^~~~~~~~~~~~~~~~
friend.cpp:32:18: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     for(int i=0;i<ad[x].size();i++)
                 ~^~~~~~~~~~~~~
friend.cpp:43:18: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     for(int i=0;i<ad[x].size();i++)
                 ~^~~~~~~~~~~~~
friend.cpp:45:18: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     for(int i=0;i<inloc[x].size();i++)
                 ~^~~~~~~~~~~~~~~~
friend.cpp:53:18: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     for(int i=0;i<v[x].size();i++)
                 ~^~~~~~~~~~~~
friend.cpp:55:18: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     for(int i=0;i<ad[x].size();i++)
                 ~^~~~~~~~~~~~~
friend.cpp:57:18: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     for(int i=0;i<inloc[x].size();i++)
                 ~^~~~~~~~~~~~~~~~
friend.cpp: In function 'int findSample(int, int*, int*, int*)':
friend.cpp:70:15: warning: unused variable 'k' [-Wunused-variable]
         int j,k,ok,mx=0;
               ^

• Subtask 1: 0 / 11
• Subtask 2: 8 / 8
• Subtask 3: 8 / 8
• Subtask 4: 19 / 19
• Subtask 5: 0 / 23
• Subtask 6: 0 / 31