Submission #742230

#	Time	Username	Problem	Language	Result	Execution time	Memory
742230		angels	Friend (IOI14_friend)	C++14	16 / 100	3 ms	2772 KiB

friend

#include <bits/stdc++.h>

using namespace std;

int N,*C,*H,*P;
vector <int> sol[100000];
bool visited[100000];

void connect(int a,int b)
{
    sol[a].push_back(b);
    sol[b].push_back(a); 
}

void prijatelstva()
{
    int i;
    for (i=1; i<N; i++)
    {
        if(P[i]==0) 
        {
            connect(H[i], i);
        }
        if(P[i]==1) 
        {
            for(int j=sol[H[i]].size(); j>0; j--)
                connect(sol[H[i]][j], i);
        }
        if(P[i]==2) 
        {
            for(int j=sol[H[i]].size(); j>0; j--)
                connect(sol[H[i]][j], i);
            connect(H[i], i);
        }
    }
}
int dfs(int x, int value)
{
    int i, rez=0;
    if(x==N)    //end
        return value;
    for (i=sol[x].size(); i--;)
    {
        if(visited[sol[x][i]]) 
            break;
    }
    if(i<0)
    {
        visited[x]=1;
        rez=dfs(x+1, value+C[x]);
    }
    visited[x]=0;
    int y=dfs(x+1, value);
    if(rez<y) 
        rez=y;
    return rez;
}

int subtask1()
{
    prijatelstva();
    return dfs(0, 0);
}

int subtask2()
{
    int zbir=0;
    for (int i=0; i<N; i++) 
        zbir+=C[i];
    return zbir;
}

int subtask3()
{
    int maxx=-1;
    for (int i=0; i<N; i++)
    {
        if(C[i]>maxx) 
            maxx=C[i];
    }
    return maxx;
}
int findSample(int n, int confidence[], int host[], int protocol[])
{
    int i;
    N=n;
    C=confidence;
    H=host; 
    P=protocol;
    if (N<=10) 
        return subtask1();
    for(i=2; i<N; i++) 
        if(protocol[i-1]!=protocol[i]) 
            break;
    if(i>=N)
    {
        if(protocol[1]==1)
        {
            return subtask2();
        }
        if(protocol[1]==2)
            return subtask3();
    }
    
    return 0;
}

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