poj 2785 4 values whose sum is 0

来源:互联网 发布:步步高v205软件java 编辑:程序博客网 时间:2024/06/05 22:55

题目大意:有四列整数,从每列中取出一个数相加,和如果是0,即满足要求,求这样的数的组合有多少种。

方法:由于每列最多可以有4000个数,所以暴力的话会超时,可以先将两列的数的所有组合的和存在两个数组中,然后看一个数组中的数在另一个数组中能否找到对应的数相加起来为0,这样时间复杂是O(16000000);

AC代码:

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<queue>
#include<time.h>
using namespace std;
int a[100005],b[100005],c[100005];
int main()
{
    int t,n,i,j,sum,suma,sumb,w;
    cin>>t;
    for(i=1;i<=t;i++)
    {
        cout<<"Case "<<i<<": ";
        cin>>n;
        suma=sumb=0;
        for(j=0;j<n;j++)
        {
            cin>>a[j];
            suma+=a[j];
        }
        for(j=0;j<n;j++)
        {
            cin>>b[j];
            sumb+=b[j];
        }
        if(suma<sumb)
        {
            cout<<"Not possible"<<endl;
            continue;
        }
        for(j=0;j<n;j++)
        {
            c[j]=a[j]-b[j];
        }
        sum=0;
        w=1;
        for(j=0;j<n;j++)
        {
           sum+=c[j];
           if(sum<0)
           {
               sum=0;
               w=j+2;
           }
        }
        cout<<"Possible from station "<<w<<endl;
    }
    return 0;
}

0 0
原创粉丝点击