인프런 커뮤니티 질문&답변

안녕하세요!

5-H 13144 문제 답을 구하는 코드가 이해되지 않아 문의드립니다. ㅠㅠ

예제 : n= 5, 1 2 3 2 1 를 넣어서 값을 구해보았는데

e=3, s=0, ret=3 / e=4, s=1, ret =6 / e=5,s=2, ret=12

이렇게 나오긴 하던데 ret 코드를 이해하진 못했어요..

그리고 연속한 1개 이상의 수를 뽑는 경우,

예제: n =5, 1 2 3 2 1

1개 뽑을 때 1,2,3 -> 3가지

2개 뽑을 때 1,2 / 1,3 / 2,3 -> 3가지

3개 뽑을 때 1,2,3 -> 1가지

총 7가지 .. 아닌가용...ㅠㅠ

그럼 시간되실 때 답변 부탁드려요..

long long s, e, cnt[100001], n, a[100001];

long long ret;

int main(){

scanf("%d", &n);

for(int i = 0; i < n; i++){

scanf("%lld", a + i);

}

while(e < n){

if(!cnt[a[e]]){

cnt[a[e]]++;

e++;

}else{

ret += (e - s);

cnt[a[s]]--;

s++;

}

}

ret += (long long)(e - s) * (e - s + 1) / 2;

printf("%lld\n", ret);

return 0;

}