글 읽기 - 어디가 틀린지 모르겠네요;;

shhj1998 7년 전 0

다른 채점 사이트에서 돌려보면 만점 나오던데 여기서는 안되네요;;

놓친 예외가 있나요??

#include<cstdio>
#include<vector>
#include<algorithm>
 
using namespace std;
 
long long hap;
long long T;
int n,m;
int A[1002];
int B[1002];
int ans;
 
bool comp(int a, int b){
    return b < a;
}
 
int main(){
    int i,j,k;
    scanf("%lld",&T);
    scanf("%d",&n);
    for(i=0;i<n;i++)scanf("%d",&A[i]);
    scanf("%d",&m);
    for(i=0;i<m;i++)scanf("%d",&B[i]);
     
    vector<int> Ahap;
    vector<int> Bhap;
    for(i=1;i<=n;i++){
        j = 0;
        hap = 0;
        for(k = 0;k<i;k++)hap+=A[k];
        if(hap <= T)Ahap.push_back(hap);
        while(j + i < n){
            hap -= A[j];
            hap += A[j + i];
            j++;
            if(hap < T)Ahap.push_back(hap);
        }
    }
    for(i=1;i<=m;i++){
        j = 0;
        hap = 0;
        for(k = 0;k<i;k++)hap+=B[k];
        if(hap <= T)Bhap.push_back(hap);
        while(j + i < m){
            hap -= B[j];
            hap += B[j + i];
            j++;
            if(hap < T)Bhap.push_back(hap);
        }
    }
    sort(Ahap.begin(), Ahap.end());
    sort(Bhap.begin(), Bhap.end(), comp);
     
    j = 0;
    for(i=0;i<Ahap.size();i++){
        while(j < Bhap.size() - 1 && Ahap[i] + Bhap[j] > T)j++;
        k = j;
        while(k < Bhap.size() && Ahap[i] + Bhap[k] == T){
            ans++;
            k++;
        }
    }
    printf("%d",ans);
     
    return 0;
}

댓글을 작성하려면 로그인해야 합니다.

문제

문제

출처

ICPC

어디가 틀린지 모르겠네요;;

shhj1998 7년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말