学渣的逆袭

Description

老师在黑板上写了四个数列a,b,c,d，数列a,b,c,d分别有i,j,k,l个数，突然间老师很生气的把正在睡觉的豆子喊了起来，问：“这是你第x次上课睡觉了！现在给你个赎罪的机会，你从每个数列中选择一个数，有多少种选法使他们的和为x？”，豆子实在太慌乱了，小伙伴们能告诉豆子正确答案吗？
Input

第一行有四个整数i,j,k,l(1<=i,j,k,l<=500)，第二行有i个数a1,a2...ai，第三行有j个数b1,b2...bj，第四行有k个数c1,c2...ck，第五行有l个数d1,d2...dl。第六行有一个数m,接下来m行询问，每行有一个数字x。(m<=10),(|x|<10^10),(|a|,|b|,|c|,|d|<=10^8)
Output

输出有m行，每行输出一个x对应的答案。
Sample Input
2 2 2 2
1 2
3 4
5 6
7 8
3
7
16
17
Sample Output
0
1
4
HINT

时间给了3s，还是挺多的，枚举+二分就可以了。

写了三个，一直在优化时间最后终于优化到1s以内了O(∩_∩)O~~。

第一个1800ms。。

#include<cstdio>
#include<cstring>
#include<stack>
#include<vector>
#include<queue>
#include<cmath>
#include<cstdlib>
#include<algorithm>
using namespace std;
const int oo = 0x3f3f3f3f;
const int maxn = 1e4+7;
const int mod = 998244353;
typedef long long LL;
int a[maxn], b[maxn], c[maxn], d[maxn];
LL ans;
int main()
{
    int i, j, q, w, e, r, m, x, k;
    while(~scanf("%d %d %d %d", &q, &w, &e, &r))
    {
        ans = 0;
        for(i = 0; i < q; i ++) scanf("%d", &a[i]);
        for(i = 0; i < w; i++)scanf("%d", &b[i]);
        for(i = 0; i < e; i++) scanf("%d", &c[i]);
        for(i = 0; i < r; i++)scanf("%d", &d[i]);
        sort(a, a+q); sort(b, b+w);
        sort(c, c+e); sort(d, d+r);
        scanf("%d", &m);
        while(m--)
        {
            scanf("%d", &x);
            ans = 0;
            for(i = 0; i < q; i ++)
            {
                if(a[i] > x) break;
                for(j = 0; j < w; j ++)
                {
                    if(a[i]+b[j] > x) break;
                    for(k = 0; k < e; k++)
                    {
                        if(a[i]+b[j]+c[k] > x) break;
                        int v = x-a[i]-b[j]-c[k];
                        if(binary_search(d, d+r, v))ans++;
                    }
                }
            }
            printf("%lld\n", ans);
        }
    }
    return 0;
}

/**************************************************************
    Problem: 1785
    User: 2759894160
    Language: C++
    Result: Accepted
    Time:1858 ms
    Memory:1032 kb
****************************************************************/

第二个 1600ms。。 只少了200不过聊胜于无，

#include<cstdio>
#include<cstring>
#include<stack>
#include<vector>
#include<queue>
#include<cmath>
#include<cstdlib>
#include<algorithm>
using namespace std;
const int oo = 0x3f3f3f3f;
const int maxn = 1e4+7;
const int mod = 998244353;
typedef long long LL;
int a[maxn], b[maxn], c[maxn], d[maxn];
LL ans;
void init(int &m, int o[], int n, int val)
{
    m = lower_bound(o, o+n, val)-o;
    if(o[m] > val || m == n) m--;
}
int main()
{
    int i, j, q, w, e, r, m, x, k;
    int z, s, v;
    while(~scanf("%d %d %d %d", &q, &w, &e, &r))
    {
        for(i = 0; i < q; i ++) scanf("%d", &a[i]);
        for(i = 0; i < w; i++)scanf("%d", &b[i]);
        for(i = 0; i < e; i++) scanf("%d", &c[i]);
        for(i = 0; i < r; i++)scanf("%d", &d[i]);
        sort(a, a+q); sort(b, b+w);
        sort(c, c+e); sort(d, d+r);
        scanf("%d", &m);
        while(m--)
        {
            scanf("%d", &x);
            ans = 0;
           init(z, a, q, x); init(s, b, w, x); init(v, c, e, x);
            for(i = 0; i <= z; i ++)
            {
                if(a[i] > x) break;
                for(j = 0; j <= s; j ++)
                {
                    if(a[i]+b[j] > x) break;
                    for(k = 0; k <= v; k++)
                    {
                        if(a[i]+b[j]+c[k] > x) break;
                        int v = x-a[i]-b[j]-c[k];
                        if(binary_search(d, d+r, v))ans++;
                    }
                }
            }
            printf("%lld\n", ans);
        }
    }
    return 0;
}

/**************************************************************
    Problem: 1785
    User: 2759894160
    Language: C++
    Result: Accepted
    Time:1667 ms
    Memory:1032 kb
****************************************************************/

第三个， 本来把后三个数组的所有和放到了一个新的数组里面， 不过500*500*500太大， 于是计划破产， 就放了2个500*500  500*500的。。（700ms 优雅的暴力）

#include<cstdio>
#include<cstring>
#include<stack>
#include<vector>
#include<queue>
#include<cmath>
#include<cstdlib>
#include<algorithm>
using namespace std;
const int oo = 0x3f3f3f3f;
const int maxn = 1e6+7;
const int mod = 998244353;
typedef long long LL;
int a[maxn], b[maxn], c[maxn], d[maxn];
int fir[maxn], sec[maxn];
LL ans;
int main()
{
    int i, j, q, w, e, r, m, x, k, cnt, cnt1;
    while(~scanf("%d %d %d %d", &q, &w, &e, &r))
    {
        cnt = cnt1 = ans = 0;
        for(i = 0; i < q; i ++) scanf("%d", &a[i]);
        for(i = 0; i < w; i++)scanf("%d", &b[i]);
        for(i = 0; i < e; i++) scanf("%d", &c[i]);
        for(i = 0; i < r; i++)scanf("%d", &d[i]);
        for(i = 0; i < q; i++)
            for(j = 0; j < w; j++)
                fir[cnt++] = a[i]+b[j];
        for(i = 0; i < e; i++)
            for(j = 0; j < r; j++)
            sec[cnt1++] = c[i]+d[j];
        sort(fir, fir+cnt); sort(sec, sec+cnt1);
        scanf("%d", &m);
        while(m--)
        {
            scanf("%d", &x);
            ans = 0;
            for(i = 0; i < cnt; i++)
            {
                int v = x-fir[i];
                int l, r;
                if(binary_search(sec, sec+cnt1, v))
                {
                    l = lower_bound(sec, sec+cnt1, v)-sec;
                    r = upper_bound(sec, sec+cnt1, v)-sec;
                    ans += r-l;
                }
            }
            printf("%lld\n", ans);
        }
    }
    return 0;
}

/**************************************************************
    Problem: 1785
    User: 2759894160
    Language: C++
    Result: Accepted
    Time:704 ms
    Memory:24312 kb
****************************************************************/