Submission #154741

#	Submission time	Handle	Problem	Language	Result	Execution time	Memory
154741	2019-09-24T11:27:58 Z	arnold518	PIN (CEOI10_pin)	C++14	100 / 100	157 ms	2908 KB

pin

#include <bits/stdc++.h>
using namespace std;

typedef long long ll;
typedef pair<int, int> pii;
typedef pair<ll, ll> pll;

const int MAXN = 5e4;

int N, D;
int A[MAXN+10][5], T[5]={1, 40, 40*40, 40*40*40};
ll arr[5], ans[5];

int f(char x)
{
    if('0'<=x && x<='9') return x-'0';
    else return x-'a'+10;
}

void solve()
{
    int i, j, k, p, q;
    for(i=0; i<4; i++)
    {
        map<int, int> M;
        for(j=1; j<=N; j++)
        {
            arr[3]+=M[A[j][4]-A[j][i]];
            M[A[j][4]-A[j][i]]++;
        }
    }

    for(i=0; i<4; i++)
    {
        for(j=i+1; j<4; j++)
        {
            map<int, int> M;
            for(k=1; k<=N; k++)
            {
                arr[2]+=M[A[k][4]-A[k][i]-A[k][j]];
                M[A[k][4]-A[k][i]-A[k][j]]++;
            }
        }
    }

    for(i=0; i<4; i++)
    {
        map<int, int> M;
        for(j=1; j<=N; j++)
        {
            arr[1]+=M[A[j][i]];
            M[A[j][i]]++;
        }
    }

    ans[1]=arr[3];
    ans[2]=arr[2]-3*arr[3];
    ans[3]=arr[1]-2*arr[2]+3*arr[3];
    ans[4]=(ll)N*(N-1)/2-ans[1]-ans[2]-ans[3];
}

int main()
{
    int i, j;

    scanf("%d%d", &N, &D);
    for(i=1; i<=N; i++)
    {
        char s[5];
        scanf("%s", s);
        A[i][0]=f(s[0])*T[0];
        A[i][1]=f(s[1])*T[1];
        A[i][2]=f(s[2])*T[2];
        A[i][3]=f(s[3])*T[3];
        A[i][4]=A[i][0]+A[i][1]+A[i][2]+A[i][3];
    }
    solve();
    printf("%lld", ans[D]);
}

Compilation message

pin.cpp: In function 'void solve()':
pin.cpp:22:18: warning: unused variable 'p' [-Wunused-variable]
     int i, j, k, p, q;
                  ^
pin.cpp:22:21: warning: unused variable 'q' [-Wunused-variable]
     int i, j, k, p, q;
                     ^
pin.cpp: In function 'int main()':
pin.cpp:64:12: warning: unused variable 'j' [-Wunused-variable]
     int i, j;
            ^
pin.cpp:66:10: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
     scanf("%d%d", &N, &D);
     ~~~~~^~~~~~~~~~~~~~~~
pin.cpp:70:14: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
         scanf("%s", s);
         ~~~~~^~~~~~~~~

Subtask #1

100.0 / 100.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	5 ms	376 KB	Output is correct
2	Correct	5 ms	376 KB	Output is correct
3	Correct	5 ms	376 KB	Output is correct
4	Correct	38 ms	948 KB	Output is correct
5	Correct	47 ms	1016 KB	Output is correct
6	Correct	46 ms	1016 KB	Output is correct
7	Correct	39 ms	888 KB	Output is correct
8	Correct	51 ms	1144 KB	Output is correct
9	Correct	77 ms	1400 KB	Output is correct
10	Correct	85 ms	1528 KB	Output is correct
11	Correct	48 ms	1072 KB	Output is correct
12	Correct	81 ms	1468 KB	Output is correct
13	Correct	53 ms	1136 KB	Output is correct
14	Correct	48 ms	1016 KB	Output is correct
15	Correct	81 ms	1400 KB	Output is correct
16	Correct	103 ms	2172 KB	Output is correct
17	Correct	157 ms	2908 KB	Output is correct
18	Correct	120 ms	2424 KB	Output is correct
19	Correct	139 ms	2580 KB	Output is correct
20	Correct	155 ms	2792 KB	Output is correct