CCF 相邻对数 C语言 python

试题

试题编号： 201409-1
试题名称： 相邻数对
时间限制： 1.0s
内存限制： 256.0MB
问题描述：

给定n个不同的整数，问这些数中有多少对整数，它们的值正好相差1。

输入格式
　　输入的第一行包含一个整数n，表示给定整数的个数。
　　第二行包含所给定的n个整数。
输出格式
　　输出一个整数，表示值正好相差1的数对的个数。

样例

样例输入
6
10 2 6 3 7 8
样例输出
3
样例说明
　　值正好相差1的数对包括(2, 3), (6, 7), (7, 8)。
评测用例规模与约定
　　1<=n<=1000，给定的整数为不超过10000的非负整数。

C程序

#include<stdio.h>

int main() {
	int n;

	scanf("%d",&n);
	
	int a[n];
	int i,j,count,temp; 
	
	
	for(i=0; i<n; i++) {
		scanf("%d",&a[i]);
	}
	
	for(i=0,count=0;i<n-1;i++){
		for(j=i;j<n;j++){
			temp = a[i]-a[j];
			if(temp==-1 || temp==1){
				count++;
			}
		}
	}
	
	printf("%d\n",count);
	return 0;
}

python

n = int(input())
data = list(map(int, input().split()))
count = 0
for i in range(0, n - 1):
    for j in range(i, n):
        temp = data[i] - data[j]
        if temp == -1 or temp == 1:
            count += 1

print(count)