2129. 礼物

题目描述

圣诞节这天，某商店准备了N个礼品盒，分别用整数1-N进行编号。其中，编号为1的盒子中有一个糖果，编号为2的盒子中有2个糖果，。。。编号为N的盒子中有N个糖果。这天一早，中山幼儿园的K个小朋友一起来到这间商店。作为当天的第一批顾客，这些小朋友可以从这N个礼品盒中选出两个拿走。小朋友们商量了一会儿后决定，他们拿走的糖果并不一定要多，但是一定要能够刚好平分给每个人。即拿走的两个盒子中的糖果总数一定要使K的倍数。现在他们想知道一共有多少种方案可供选择。

输入

每行两个正整数N和K，其中1<=N<=$10^9$,$1<=K<=10^9$。一行N=K=0
表示输入结束，这一行不用处理。

输出

对输入中除了N=K=0外的每一行，输出一行，这一行只有一个数，即其相对应的输入所得到的方案数。

数据范围限制

20%的数据N<=100;
80%的数据K<=1000;
每个输入文件最多有200行输入数据。

Code

20分

#include<iostream>
#include<stdio.h>
#pragma GCC optimize (3)
using namespace std;
int main()
{
	//freopen("gift.in","r",stdin);
	//freopen("gift.out","w",stdout);
	long long n,k;
	long long ans;
	long long sum;
	for(;;)
	{
		cin >> n >> k;
		if(n != 0 && n != 1)
		{
			ans = 0;
			for(register long long i = 1;i <= n;i++)
				for(register long long j = n;j >= i + 1;j--)
				{
					sum = i + j;
					if(sum % k == 0 && sum != 1)
						ans++;
				}
				cout << ans << endl;
		}
		else if(n == 1)
			cout << 0 << endl;	
		else 
			break;
	}
	return 0;
}

100分

#include<cstdio>
#include<iostream>
using namespace std;
long long n,m,x,y,c,s;
int main()
{
	freopen("gift.in","r",stdin);
	freopen("gift.out","w",stdout);
	for(;;)
	{
		cin >> n >> m;
		if(n && m)
		{
			x = n / m;
			y = n % m;
			c = (m - 1) / 2;
			s = x * (x - 1) / 2 + x * x * c;
			if(m % 2 == 0)
				s += x * (x - 1) / 2;
			if(y > 0)
			{
				if(y >= c)
					s = s + x * c;
				else
					s = s + x * y;
				y -= c;
				if(y > 0 && m % 2 == 0)
				{
					s = s - x * (x - 1) / 2 + x * (x + 1) / 2;
					y--;
				}
			}
			if(y > 0)
				s += (x + 1) * y;
			cout << s << endl;
		}
		else
			break;
	}
}